diff --git a/config.json b/config.json
index 418b364..b57fe5d 100644
--- a/config.json
+++ b/config.json
@@ -1,36 +1,31 @@
 {
+  "activation": "gelu",
   "architectures": [
     "DistilBertForSequenceClassification"
   ],
-  "activation": "gelu",
   "attention_dropout": 0.1,
   "dim": 768,
   "dropout": 0.1,
   "finetuning_task": "sst-2",
   "hidden_dim": 3072,
+  "id2label": {
+    "0": "NEGATIVE",
+    "1": "POSITIVE"
+  },
   "initializer_range": 0.02,
-  "is_decoder": false,
-  "max_position_embeddings": 512,
-  "n_heads": 12,
-  "n_layers": 6,
-  "num_labels": 2,
-  "output_attentions": false,
-  "output_hidden_states": false,
-  "output_past": true,
-  "pruned_heads": {},
-  "qa_dropout": 0.1,
-  "seq_classif_dropout": 0.2,
-  "sinusoidal_pos_embds": false,
-  "tie_weights_": true,
-  "torchscript": false,
-  "use_bfloat16": false,
-  "vocab_size": 30522,
   "label2id": {
     "NEGATIVE": 0,
     "POSITIVE": 1
   },
-  "id2label": {
-    "0": "NEGATIVE",
-    "1": "POSITIVE"
-  }
+  "max_position_embeddings": 512,
+  "model_type": "distilbert",
+  "n_heads": 12,
+  "n_layers": 6,
+  "output_past": true,
+  "pad_token_id": 0,
+  "qa_dropout": 0.1,
+  "seq_classif_dropout": 0.2,
+  "sinusoidal_pos_embds": false,
+  "tie_weights_": true,
+  "vocab_size": 30522
 }