polieste
/

fastAbs_large

text2text-generation

text-generation-inference

Model card Files Files and versions

polieste commited on Jan 10, 2024

Commit

a0b942b

·

1 Parent(s): 42d595a

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -28,7 +28,7 @@ encoding = tokenizer(text, return_tensors="pt")
 input_ids, attention_masks = encoding["input_ids"].to("cuda"), encoding["attention_mask"].to("cuda")
 outputs = model.generate(
     input_ids=input_ids, attention_mask=attention_masks,
-    max_length=128,
     early_stopping=True
 )
 for output in outputs:

 input_ids, attention_masks = encoding["input_ids"].to("cuda"), encoding["attention_mask"].to("cuda")
 outputs = model.generate(
     input_ids=input_ids, attention_mask=attention_masks,
+    max_length=512,
     early_stopping=True
 )
 for output in outputs: