polieste
/

fastAbs_base

text2text-generation

Model card Files Files and versions

polieste commited on Jan 8, 2024

Commit

e483602

·

1 Parent(s): ab1ce0e

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -131,7 +131,7 @@ input_ids = tokenizer(
 output_ids = model.generate(
     input_ids=input_ids,
-    max_length=256,
     no_repeat_ngram_size=2,
     num_beams=4
 )[0]

 output_ids = model.generate(
     input_ids=input_ids,
+    max_length=128,
     no_repeat_ngram_size=2,
     num_beams=4
 )[0]