Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -40,7 +40,7 @@ def predict(message, history, tokens, temperature, language):
|
|
40 |
history_transformer_format = history + [[message, ""]]
|
41 |
stop = StopOnTokens()
|
42 |
|
43 |
-
messages = "".join(["".join(["\n<english>:"+item[0]+"
|
44 |
for item in history_transformer_format])
|
45 |
|
46 |
model_inputs = tokenizer([messages], return_tensors="pt").to("cuda")
|
|
|
40 |
history_transformer_format = history + [[message, ""]]
|
41 |
stop = StopOnTokens()
|
42 |
|
43 |
+
messages = "".join(["".join(["\n<english>:"+item[0]+"<NL>\n", tag+item[1]])
|
44 |
for item in history_transformer_format])
|
45 |
|
46 |
model_inputs = tokenizer([messages], return_tensors="pt").to("cuda")
|