EbanLee
/

kobart-title

Text2Text Generation

Inference Endpoints

Model card Files Files and versions Community

EbanLee commited on Mar 15

Commit

c17e2b2

•

1 Parent(s): 3e7a3ac

Update README.md

Files changed (1) hide show

README.md +3 -3

README.md CHANGED Viewed

@@ -28,11 +28,11 @@ summary_text_ids = model.generate(
 input_ids=input_ids,
 bos_token_id=model.config.bos_token_id,
 eos_token_id=model.config.eos_token_id,
-length_penalty=0.8,
-max_length=30,
 min_length=3,
 num_beams=6,
-repetition_penalty=2.0,
 )
 # Decoding Text Ids

 input_ids=input_ids,
 bos_token_id=model.config.bos_token_id,
 eos_token_id=model.config.eos_token_id,
+length_penalty=1.0,
+max_length=40,
 min_length=3,
 num_beams=6,
+repetition_penalty=1.5,
 )
 # Decoding Text Ids