Update config.json

config.json

@@ -2,7 +2,7 @@
   "activation_dropout": 0.1,
   "activation_function": "gelu",
   "architectures": [
-    "ProphetNetForConditionalGeneration"
+    "XLMProphetNetForConditionalGeneration"
   ],
   "attention_dropout": 0.1,
   "bos_token_id": 2,
@@ -28,9 +28,9 @@
     "LABEL_1": 1,
     "LABEL_2": 2
   },
-  "max_length": 256,
+  "max_length": 512,
   "max_position_embeddings": 512,
-  "model_type": "prophetnet",
+  "model_type": "xlm-prophetnet",
   "ngram": 2,
   "num_buckets": 32,
   "num_decoder_attention_heads": 16,
@@ -44,10 +44,10 @@
   "task_specific_params": {
     "summarization": {
       "early_stopping": true,
-      "length_penalty": 1.0,
+      "length_penalty": 2.0,
       "no_repeat_ngram_size": 3,
-      "num_beams": 10
+      "num_beams": 4
     }
   },
   "vocab_size": 250012
-}
+}