aixsatoshi
commited on
Commit
•
65ac526
1
Parent(s):
764ca7f
Update README.md
Browse files
README.md
CHANGED
@@ -92,6 +92,7 @@ def predict(message, history, tokens, temperature, language):
|
|
92 |
do_sample=True,
|
93 |
top_p=0.95,
|
94 |
top_k=20,
|
|
|
95 |
num_beams=1,
|
96 |
stopping_criteria=StoppingCriteriaList([stop])
|
97 |
)
|
@@ -198,6 +199,7 @@ def predict(message, tokens, temperature, language):
|
|
198 |
do_sample=True,
|
199 |
top_p=0.95,
|
200 |
top_k=20,
|
|
|
201 |
num_beams=1,
|
202 |
eos_token_id=tokenizer.eos_token_id
|
203 |
)
|
|
|
92 |
do_sample=True,
|
93 |
top_p=0.95,
|
94 |
top_k=20,
|
95 |
+
repetition_penalty=1.15,
|
96 |
num_beams=1,
|
97 |
stopping_criteria=StoppingCriteriaList([stop])
|
98 |
)
|
|
|
199 |
do_sample=True,
|
200 |
top_p=0.95,
|
201 |
top_k=20,
|
202 |
+
repetition_penalty=1.15,
|
203 |
num_beams=1,
|
204 |
eos_token_id=tokenizer.eos_token_id
|
205 |
)
|