Update config.json

config.json

@@ -1,48 +1,22 @@
 {
-  "_num_labels": 9,
+  "activation": "gelu",
   "architectures": [
-    "BertForTokenClassification"
+    "DistilBertForQuestionAnswering"
   ],
-  "attention_probs_dropout_prob": 0.1,
-  "directionality": "bidi",
-  "hidden_act": "gelu",
-  "hidden_dropout_prob": 0.1,
-  "hidden_size": 2,
-  "id2label": {
-    "0": "O",
-    "1": "B-MISC",
-    "2": "I-MISC",
-    "3": "B-PER",
-    "4": "I-PER",
-    "5": "B-ORG",
-    "6": "I-ORG",
-    "7": "B-LOC",
-    "8": "I-LOC"
-  },
+  "attention_dropout": 0.1,
+  "dim": 2,
+  "dropout": 0.1,
+  "hidden_dim": 2,
   "initializer_range": 0.02,
-  "intermediate_size": 4,
-  "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": 2,
-  "num_hidden_layers": 2,
+  "model_type": "distilbert",
+  "n_heads": 2,
+  "n_layers": 2,
+  "output_past": true,
   "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,
+  "qa_dropout": 0.1,
+  "seq_classif_dropout": 0.2,
+  "sinusoidal_pos_embds": true,
+  "tie_weights_": true,
   "vocab_size": 28996
 }