Training done

config.json

@@ -50,7 +50,7 @@
       "LABEL_1": 1
     },
     "length_penalty": 1.0,
-    "max_length": 768,
+    "max_length": 1024,
     "max_position_embeddings": 1536,
     "min_length": 0,
     "model_type": "mbart",
@@ -83,13 +83,13 @@
     "top_p": 1.0,
     "torch_dtype": null,
     "torchscript": false,
-    "transformers_version": "4.29.1",
+    "transformers_version": "4.29.2",
     "typical_p": 1.0,
     "use_bfloat16": false,
     "use_cache": true,
-    "vocab_size": 57580
+    "vocab_size": 57533
   },
-  "decoder_start_token_id": 57579,
+  "decoder_start_token_id": 57526,
   "encoder": {
     "_name_or_path": "",
     "add_cross_attention": false,
@@ -126,8 +126,8 @@
       "1": "LABEL_1"
     },
     "image_size": [
-      1280,
-      960
+      560,
+      560
     ],
     "initializer_range": 0.02,
     "is_decoder": false,
@@ -180,7 +180,7 @@
     "top_p": 1.0,
     "torch_dtype": null,
     "torchscript": false,
-    "transformers_version": "4.29.1",
+    "transformers_version": "4.29.2",
     "typical_p": 1.0,
     "use_absolute_embeddings": false,
     "use_bfloat16": false,

generation_config.json

@@ -1,10 +1,10 @@
 {
   "_from_model_config": true,
   "bos_token_id": 0,
-  "decoder_start_token_id": 57579,
+  "decoder_start_token_id": 57526,
   "eos_token_id": 2,
   "forced_eos_token_id": 2,
-  "max_length": 768,
+  "max_length": 1024,
   "pad_token_id": 1,
-  "transformers_version": "4.29.1"
+  "transformers_version": "4.29.2"
 }

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:bbe3acd17c0ccda4227f16f3cae5abfb8aa8d01c40b3c48ea01826fa1580d31f
-size 809404185
+oid sha256:808c375062d053aa9f3801dfb2cf8bb43d67ed770f5c0941a5ccef4487faaa40
+size 809211673