Upload MBartForConditionalGeneration

config.json

@@ -19,7 +19,7 @@
   "decoder_layers": 12,
   "decoder_start_token_id": 2,
   "dropout": 0.1,
-  "early_stopping": null,
+  "early_stopping": true,
   "encoder_attention_heads": 16,
   "encoder_ffn_dim": 4096,
   "encoder_layerdrop": 0.0,
@@ -39,12 +39,12 @@
     "LABEL_1": 1,
     "LABEL_2": 2
   },
-  "max_length": null,
+  "max_length": 200,
   "max_position_embeddings": 1024,
   "model_type": "mbart",
   "normalize_before": true,
   "normalize_embedding": true,
-  "num_beams": null,
+  "num_beams": 5,
   "num_hidden_layers": 12,
   "output_past": true,
   "pad_token_id": 1,
@@ -52,7 +52,7 @@
   "static_position_embeddings": false,
   "tokenizer_class": "MBart50Tokenizer",
   "torch_dtype": "float32",
-  "transformers_version": "4.45.2",
+  "transformers_version": "4.44.2",
   "use_cache": true,
   "vocab_size": 250054
 }

generation_config.json

@@ -8,5 +8,5 @@
   "max_new_tokens": 16,
   "num_beams": 5,
   "pad_token_id": 1,
-  "transformers_version": "4.45.2"
+  "transformers_version": "4.44.2"
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:37a7c878186b50edf7f55492719cddc277fc7f3cbba8f1a81b012d40beaca798
+oid sha256:ef0d81dcef2a022f55afa6487e68eed29ff5e77a17b01cf01be952dc85834cfb
 size 2444578688