diff --git a/config.json b/config.json
index 9b15f3b..90bf203 100644
--- a/config.json
+++ b/config.json
@@ -1,6 +1,10 @@
 {
   "activation_dropout": 0.0,
-  "architectures": null,
+  "architectures": [
+    "BartModel",
+    "BartForMaskedLM",
+    "BartForSequenceClassification"
+  ],
   "attention_dropout": 0.0,
   "bos_token_id": 0,
   "classif_dropout": 0.0,
@@ -33,11 +37,11 @@
   "model_type": "bart",
   "num_beams": 1,
   "num_hidden_layers": 12,
-  "num_labels": 3,
+  "num_labels": 2,
   "num_return_sequences": 1,
   "output_attentions": false,
   "output_hidden_states": false,
-  "output_past": true,
+  "output_past": false,
   "pad_token_id": 1,
   "pruned_heads": {},
   "repetition_penalty": 1.0,