Update README.md

README.md

@@ -92,6 +92,7 @@ def predict(message, history, tokens, temperature, language):
         do_sample=True,
         top_p=0.95,
         top_k=20,
+        repetition_penalty=1.15,
         num_beams=1,
         stopping_criteria=StoppingCriteriaList([stop])
     )
@@ -198,6 +199,7 @@ def predict(message, tokens, temperature, language):
         do_sample=True,
         top_p=0.95,
         top_k=20,
+        repetition_penalty=1.15,
         num_beams=1,
         eos_token_id=tokenizer.eos_token_id
     )