Update app.py
Browse files
app.py
CHANGED
@@ -34,10 +34,10 @@ def local_generate(
|
|
34 |
prompt,
|
35 |
generation_config,
|
36 |
max_new_tokens,
|
37 |
-
do_sample,
|
38 |
-
top_p,
|
39 |
-
repetition_penalty,
|
40 |
-
temperature=
|
41 |
):
|
42 |
response_output = generator(
|
43 |
prompt,
|
|
|
34 |
prompt,
|
35 |
generation_config,
|
36 |
max_new_tokens,
|
37 |
+
do_sample=True,
|
38 |
+
top_p=0.25,
|
39 |
+
repetition_penalty=1.2,
|
40 |
+
temperature=1.0
|
41 |
):
|
42 |
response_output = generator(
|
43 |
prompt,
|