X-Wang
/

pruned-mt5-small

Text2Text Generation

Generated from Trainer

Inference Endpoints

Model card Files Files and versions Community

X-Wang commited on Jul 28, 2023

Commit

bd9b6aa

•

1 Parent(s): 7101cf1

Upload MT5ForConditionalGeneration

Files changed (3) hide show

config.json +2 -2
generation_config.json +4 -3
pytorch_model.bin +2 -2

config.json CHANGED Viewed

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

 {
+  "_name_or_path": "/root/autodl-tmp/trained/checkpoint-245000",
   "architectures": [
     "MT5ForConditionalGeneration"
   ],
   "tie_word_embeddings": false,
   "tokenizer_class": "T5Tokenizer",
   "torch_dtype": "float32",
+  "transformers_version": "4.30.2",
   "use_cache": true,
   "vocab_size": 40475
 }

generation_config.json CHANGED Viewed

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

   "decoder_start_token_id": 0,
   "do_sample": true,
   "eos_token_id": 1,
+  "length_penalty": 1.5,
+  "max_length": 128,
+  "num_beams": 5,
   "pad_token_id": 0,
+  "transformers_version": "4.30.2"
 }

pytorch_model.bin CHANGED Viewed

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

 version https://git-lfs.github.com/spec/v1
+oid sha256:c1f2bf983b1736e181dfaf878cb8b4466c786cc9e3d087f921f6dbde021fe6b3
+size 342094917