Upload MT5ForConditionalGeneration

config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "X-Wang/pruned-mt5-small",
+  "_name_or_path": "/root/autodl-tmp/trained/checkpoint-245000",
   "architectures": [
     "MT5ForConditionalGeneration"
   ],
@@ -25,7 +25,7 @@
   "tie_word_embeddings": false,
   "tokenizer_class": "T5Tokenizer",
   "torch_dtype": "float32",
-  "transformers_version": "4.31.0",
+  "transformers_version": "4.30.2",
   "use_cache": true,
   "vocab_size": 40475
 }

generation_config.json

@@ -2,8 +2,9 @@
   "decoder_start_token_id": 0,
   "do_sample": true,
   "eos_token_id": 1,
-  "max_length": 512,
-  "num_beams": 6,
+  "length_penalty": 1.5,
+  "max_length": 128,
+  "num_beams": 5,
   "pad_token_id": 0,
-  "transformers_version": "4.31.0"
+  "transformers_version": "4.30.2"
 }

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:f6bc2bbee2787175d0247bb0937d9ab89908914a1876d5975f85b3bcff5010e1
-size 342099461
+oid sha256:c1f2bf983b1736e181dfaf878cb8b4466c786cc9e3d087f921f6dbde021fe6b3
+size 342094917