IlyaGusev
/

rut5_base_sum_gazeta

Summarization Transformers PyTorch

Russian t5 text2text-generation Inference Endpoints text-generation-inference

Model card Files Files and versions Community

IlyaGusev commited on Nov 23, 2021

Commit

27ed10e

•

1 Parent(s): 9d15e61

Update README.md

Files changed (1) hide show

README.md +1 -2

README.md CHANGED Viewed

@@ -52,8 +52,7 @@ input_ids = tokenizer(
 output_ids = model.generate(
     input_ids=input_ids,
-    no_repeat_ngram_size=3,
-    early_stopping=True
 )[0]
 summary = tokenizer.decode(output_ids, skip_special_tokens=True)

 output_ids = model.generate(
     input_ids=input_ids,
+    repetition_penalty=2.0
 )[0]
 summary = tokenizer.decode(output_ids, skip_special_tokens=True)