Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -37,7 +37,7 @@ def generate_text(
|
|
37 |
top_k=40,
|
38 |
repeat_penalty=1.2,
|
39 |
max_tokens=max_tokens,
|
40 |
-
stop=["<|im_end|>"],
|
41 |
stream=True,
|
42 |
|
43 |
)
|
|
|
37 |
top_k=40,
|
38 |
repeat_penalty=1.2,
|
39 |
max_tokens=max_tokens,
|
40 |
+
stop=["<|im_end|>", "[UNUSED_TOKEN_145]"],
|
41 |
stream=True,
|
42 |
|
43 |
)
|