surkovvv commited on
Commit
d8257a2
·
1 Parent(s): e4978be

new message format2

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -25,7 +25,7 @@ def predict(message, history):
25
  # messages = "".join(["".join(["<|start_header_id|>user<|end_header_id|>\n"+item[0],
26
  # "<|eot_id|><|start_header_id|>assistant<|end_header_id|>\n"+item[1]])
27
  # for item in history_transformer_format])
28
- messages = [{"role": item[0], "user": item[1]} for item in history_transformer_format]
29
  print(messages)
30
 
31
  # model_inputs = tokenizer([messages], return_tensors="pt") # .to("cuda")
 
25
  # messages = "".join(["".join(["<|start_header_id|>user<|end_header_id|>\n"+item[0],
26
  # "<|eot_id|><|start_header_id|>assistant<|end_header_id|>\n"+item[1]])
27
  # for item in history_transformer_format])
28
+ messages = [{"user": item[0], "content": item[1]} for item in history_transformer_format]
29
  print(messages)
30
 
31
  # model_inputs = tokenizer([messages], return_tensors="pt") # .to("cuda")