nafisehNik
/

mt5-persian-summary

text2text-generation

Inference Endpoints

Model card Files Files and versions Community

nafisehNik commited on Aug 21, 2023

Commit

a2005df

•

1 Parent(s): 3e3c7ff

Update config.json

Files changed (1) hide show

config.json +7 -2

config.json CHANGED Viewed

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "google/mt5-small",
   "architectures": [
     "MT5ForConditionalGeneration"
   ],
@@ -27,5 +27,10 @@
   "torch_dtype": "float32",
   "transformers_version": "4.21.1",
   "use_cache": true,
-  "vocab_size": 250112
 }

 {
+  "_name_or_path": "nafisehNik/mt5-persian-summary",
   "architectures": [
     "MT5ForConditionalGeneration"
   ],
   "torch_dtype": "float32",
   "transformers_version": "4.21.1",
   "use_cache": true,
+  "vocab_size": 250112,
+  "num_beams": 2,
+  "repetition_penalty": 1.0,
+  "length_penalty": 2.0,
+  "early_stopping": true,
+  "max_length": 120
 }