diff --git a/config.json b/config.json
index cb5c8d4..475108a 100644
--- a/config.json
+++ b/config.json
@@ -1,42 +1,13 @@
 {
+  "_num_labels": 9,
   "architectures": [
     "BertForTokenClassification"
   ],
   "attention_probs_dropout_prob": 0.1,
   "directionality": "bidi",
-  "finetuning_task": null,
   "hidden_act": "gelu",
   "hidden_dropout_prob": 0.1,
   "hidden_size": 1024,
-  "initializer_range": 0.02,
-  "intermediate_size": 4096,
-  "layer_norm_eps": 1e-12,
-  "max_position_embeddings": 512,
-  "num_attention_heads": 16,
-  "num_hidden_layers": 24,
-  "num_labels": 9,
-  "output_attentions": false,
-  "output_hidden_states": false,
-  "pooler_fc_size": 768,
-  "pooler_num_attention_heads": 12,
-  "pooler_num_fc_layers": 3,
-  "pooler_size_per_head": 128,
-  "pooler_type": "first_token_transform",
-  "pruned_heads": {},
-  "torchscript": false,
-  "type_vocab_size": 2,
-  "vocab_size": 28996, 
-  "label2id": {
-    "O": 0,
-    "B-MISC": 1,
-    "I-MISC": 2,
-    "B-PER": 3,
-    "I-PER": 4,
-    "B-ORG": 5,
-    "I-ORG": 6,
-    "B-LOC": 7,
-    "I-LOC": 8
-  },
   "id2label": {
     "0": "O",
     "1": "B-MISC",
@@ -47,5 +18,31 @@
     "6": "I-ORG",
     "7": "B-LOC",
     "8": "I-LOC"
-  }
+  },
+  "initializer_range": 0.02,
+  "intermediate_size": 4096,
+  "label2id": {
+    "B-LOC": 7,
+    "B-MISC": 1,
+    "B-ORG": 5,
+    "B-PER": 3,
+    "I-LOC": 8,
+    "I-MISC": 2,
+    "I-ORG": 6,
+    "I-PER": 4,
+    "O": 0
+  },
+  "layer_norm_eps": 1e-12,
+  "max_position_embeddings": 512,
+  "model_type": "bert",
+  "num_attention_heads": 16,
+  "num_hidden_layers": 24,
+  "pad_token_id": 0,
+  "pooler_fc_size": 768,
+  "pooler_num_attention_heads": 12,
+  "pooler_num_fc_layers": 3,
+  "pooler_size_per_head": 128,
+  "pooler_type": "first_token_transform",
+  "type_vocab_size": 2,
+  "vocab_size": 28996
 }