Update config.json

config.json

@@ -4,6 +4,9 @@
   "activation_function": "gelu",
   "add_bias_logits": false,
   "add_final_layer_norm": true,
+  "architectures": [
+    "BartForConditionalGeneration"
+  ],
   "attention_dropout": 0.0,
   "bos_token_id": 0,
   "classif_dropout": 0.0,
@@ -18,7 +21,6 @@
   "encoder_layerdrop": 0.0,
   "encoder_layers": 12,
   "eos_token_id": 2,
-  "extra_pos_embeddings": 2,
   "id2label": {
     "0": "LABEL_0",
     "1": "LABEL_1",
@@ -31,14 +33,13 @@
     "LABEL_1": 1,
     "LABEL_2": 2
   },
-  "max_length": 1000,
-  "decoder_start_token_id": 250020,
   "max_position_embeddings": 1024,
   "model_type": "mbart",
   "normalize_before": true,
   "variant": "prelayernorm",
   "normalize_embedding": true,
   "num_beams": 5,
+  "max_length": 1000,
   "num_hidden_layers": 12,
   "output_past": true,
   "pad_token_id": 1,