aixsatoshi commited on
Commit
ce1c0e3
·
verified ·
1 Parent(s): 715f8d8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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]+"</english>\n", tag+item[1]])
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")