EbanLee
/

kobart-title

Text2Text Generation

Inference Endpoints

Model card Files Files and versions Community

EbanLee commited on Mar 15, 2024

Commit

3e7a3ac

·

1 Parent(s): 5c9b647

config modify

Files changed (1) hide show

config.json +3 -3

config.json CHANGED Viewed

@@ -51,11 +51,11 @@
   "tokenizer_class": "PreTrainedTokenizerFast",
   "task_specific_params": {
     "summarization": {
-      "length_penalty": 0.8,
-      "max_length": 30,
       "min_length": 3,
       "num_beams": 6,
-      "repetition_penalty": 2.0
     }
   },
   "torch_dtype": "float32",

   "tokenizer_class": "PreTrainedTokenizerFast",
   "task_specific_params": {
     "summarization": {
+      "length_penalty": 1.0,
+      "max_length": 40,
       "min_length": 3,
       "num_beams": 6,
+      "repetition_penalty": 1.5
     }
   },
   "torch_dtype": "float32",