SaranaAbidueva
/

mbart50_ru_bua

text2text-generation

Model card Files Files and versions

SaranaAbidueva commited on Aug 13, 2023

Commit

48d4e07

·

1 Parent(s): 2cf4c23

Update README.md

Files changed (1) hide show

README.md +1 -2

README.md CHANGED Viewed

@@ -25,8 +25,7 @@ def translate(text, max_length=200, num_beams=5, repetition_penalty=5.0, **kwarg
         **encoded.to(model.device),
         max_length=max_length,
         num_beams=num_beams,
-        repetition_penalty=repetition_penalty,
-        # early_stopping=True,
     )
     return tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)[0]

         **encoded.to(model.device),
         max_length=max_length,
         num_beams=num_beams,
+        repetition_penalty=repetition_penalty
     )
     return tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)[0]