Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -80,7 +80,7 @@ def stream_chat(
|
|
80 |
generate_kwargs = dict(
|
81 |
input_ids=input_ids,
|
82 |
max_new_tokens = max_new_tokens,
|
83 |
-
do_sample = False,
|
84 |
top_p = top_p,
|
85 |
top_k = top_k,
|
86 |
eos_token_id = 128039,
|
|
|
80 |
generate_kwargs = dict(
|
81 |
input_ids=input_ids,
|
82 |
max_new_tokens = max_new_tokens,
|
83 |
+
do_sample = False if temperature == 0 else True,
|
84 |
top_p = top_p,
|
85 |
top_k = top_k,
|
86 |
eos_token_id = 128039,
|