Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -29,6 +29,7 @@ def run_generation(user_text, top_p, temperature, top_k, max_new_tokens, use_gen
|
|
29 |
streamer=streamer,
|
30 |
max_new_tokens=max_new_tokens,
|
31 |
do_sample=True,
|
|
|
32 |
top_p=top_p,
|
33 |
temperature=float(temperature),
|
34 |
top_k=top_k
|
|
|
29 |
streamer=streamer,
|
30 |
max_new_tokens=max_new_tokens,
|
31 |
do_sample=True,
|
32 |
+
num_beams=1,
|
33 |
top_p=top_p,
|
34 |
temperature=float(temperature),
|
35 |
top_k=top_k
|