diff --git a/config.json b/config.json
index e64ea77..ce6e7be 100644
--- a/config.json
+++ b/config.json
@@ -4,23 +4,22 @@
   ],
   "attn_type": "bi",
   "bi_data": false,
+  "bos_token_id": 1,
   "clamp_len": -1,
   "d_head": 64,
   "d_inner": 3072,
   "d_model": 768,
   "dropout": 0.1,
   "end_n_top": 5,
+  "eos_token_id": 2,
   "ff_activation": "gelu",
-  "finetuning_task": null,
   "initializer_range": 0.02,
   "layer_norm_eps": 1e-12,
   "mem_len": null,
+  "model_type": "xlnet",
   "n_head": 12,
   "n_layer": 12,
-  "n_token": 32000,
-  "num_labels": 2,
-  "output_attentions": false,
-  "output_hidden_states": false,
+  "pad_token_id": 5,
   "reuse_len": null,
   "same_length": false,
   "start_n_top": 5,
@@ -33,6 +32,6 @@
       "max_length": 210
     }
   },
-  "torchscript": false,
-  "untie_r": true
-}
\ No newline at end of file
+  "untie_r": true,
+  "vocab_size": 32000
+}