Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
|
@@ -22,7 +22,7 @@ if input_text:
|
|
| 22 |
for token in llm(
|
| 23 |
model_input,
|
| 24 |
max_tokens=32,
|
| 25 |
-
stop=[""],
|
| 26 |
echo=True,
|
| 27 |
stream=True # Enable streaming
|
| 28 |
):
|
|
|
|
| 22 |
for token in llm(
|
| 23 |
model_input,
|
| 24 |
max_tokens=32,
|
| 25 |
+
stop=["<end_of_turn>"],
|
| 26 |
echo=True,
|
| 27 |
stream=True # Enable streaming
|
| 28 |
):
|