diff --git a/config.json b/config.json
index 0ffe67a..0959743 100644
--- a/config.json
+++ b/config.json
@@ -1,18 +1,25 @@
 {
+  "afn": "gelu",
   "architectures": [
     "OpenAIGPTLMHeadModel"
   ],
-  "afn": "gelu",
   "attn_pdrop": 0.1,
   "embd_pdrop": 0.1,
   "initializer_range": 0.02,
   "layer_norm_epsilon": 1e-05,
+  "model_type": "openai-gpt",
   "n_ctx": 512,
   "n_embd": 768,
   "n_head": 12,
   "n_layer": 12,
   "n_positions": 512,
   "n_special": 0,
+  "predict_special_tokens": true,
   "resid_pdrop": 0.1,
+  "summary_activation": null,
+  "summary_first_dropout": 0.1,
+  "summary_proj_to_labels": true,
+  "summary_type": "cls_index",
+  "summary_use_proj": true,
   "vocab_size": 40478
 }