Update README.md

README.md

@@ -33,6 +33,8 @@ with torch.no_grad():
         max_new_tokens=64,
         do_sample=True,
         temperature=0.7,
+        top_p=0.9,
+        repetition_penalty=1.05,
         pad_token_id=tokenizer.pad_token_id,
     )