Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -39,7 +39,7 @@ def check_model_status():
|
|
39 |
|
40 |
def chat(message, history, temperature, max_new_tokens):
|
41 |
global model, tokenizer
|
42 |
-
stop_tokens = ["
|
43 |
|
44 |
# Ensure the model is loaded before proceeding
|
45 |
if not check_model_status():
|
|
|
39 |
|
40 |
def chat(message, history, temperature, max_new_tokens):
|
41 |
global model, tokenizer
|
42 |
+
stop_tokens = ["|im_end|"]
|
43 |
|
44 |
# Ensure the model is loaded before proceeding
|
45 |
if not check_model_status():
|