Update README.md

README.md

@@ -33,14 +33,13 @@ with torch.no_grad():
     out = model.generate(input_ids, 
                         do_sample=True,
                         num_beams=4,
-#                       temperature=1.4,
+                        temperature=1.1,
                         top_p=0.9,
                         top_k=50,
                         max_length=250,
                         min_length=50,
                         early_stopping=True,
-                        num_return_sequences=3, 
-                        no_repeat_ngram_size=3
+                        no_repeat_ngram_size=2
                         )
 
 generated_text = list(map(tokenizer.decode, out))[0]