Spaces:
Running
Running
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():
|