Nehc
/

mT5_ru_XLSum

@@ -105,7 +105,7 @@ output_ids = model.generate(
     input_ids=input_ids,
     max_length=84,
     no_repeat_ngram_size=2,
-    num_beams=5
 )[0]
 summary = tokenizer.decode(

     input_ids=input_ids,
     max_length=84,
     no_repeat_ngram_size=2,
+    num_beams=3
 )[0]
 summary = tokenizer.decode(