Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -74,6 +74,7 @@ def stream_chat(
|
|
| 74 |
top_k = top_k,
|
| 75 |
temperature = temperature,
|
| 76 |
streamer=streamer,
|
|
|
|
| 77 |
)
|
| 78 |
|
| 79 |
with torch.no_grad():
|
|
|
|
| 74 |
top_k = top_k,
|
| 75 |
temperature = temperature,
|
| 76 |
streamer=streamer,
|
| 77 |
+
pad_token_id = 10,
|
| 78 |
)
|
| 79 |
|
| 80 |
with torch.no_grad():
|