Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -81,8 +81,7 @@ def stream_chat(
|
|
81 |
temperature = temperature,
|
82 |
streamer=streamer,
|
83 |
repetition_penalty=penalty,
|
84 |
-
|
85 |
-
eos_token_id = 50279,
|
86 |
)
|
87 |
|
88 |
with torch.no_grad():
|
|
|
81 |
temperature = temperature,
|
82 |
streamer=streamer,
|
83 |
repetition_penalty=penalty,
|
84 |
+
eos_token_id = [2, 73440],
|
|
|
85 |
)
|
86 |
|
87 |
with torch.no_grad():
|