diff --git a/README.md b/README.md index 3127fb2..09e2cf0 100644 --- a/README.md +++ b/README.md @@ -76,7 +76,7 @@ outputs = pipeline( do_sample=True, temperature=0.6, top_p=1, - repetition_penalty=1.1 + repetition_penalty=1.15 ) print(outputs[0]["generated_text"][len(prompt):]) ```