Update app.py
Browse files
app.py
CHANGED
@@ -27,7 +27,7 @@ def generate(
|
|
27 |
temperature=temperature,
|
28 |
max_new_tokens=max_new_tokens,
|
29 |
top_p=top_p,
|
30 |
-
|
31 |
repetition_penalty=repetition_penalty,
|
32 |
do_sample=True,
|
33 |
seed=42,
|
|
|
27 |
temperature=temperature,
|
28 |
max_new_tokens=max_new_tokens,
|
29 |
top_p=top_p,
|
30 |
+
top_k=top_k,
|
31 |
repetition_penalty=repetition_penalty,
|
32 |
do_sample=True,
|
33 |
seed=42,
|