This commit is contained in:
parent
0a0fd07d2e
commit
9136fc2015
|
@ -67,8 +67,7 @@ class Qwen:
|
||||||
prompt += f"<|im_start|>system\n{message['text']}<|im_end|>\n\n"
|
prompt += f"<|im_start|>system\n{message['text']}<|im_end|>\n\n"
|
||||||
elif message["role"] == "user":
|
elif message["role"] == "user":
|
||||||
prompt += f"<|im_start|>user\n{message['text']}<|im_end|>\n\n"
|
prompt += f"<|im_start|>user\n{message['text']}<|im_end|>\n\n"
|
||||||
#elif message["role"] == "assistant":
|
prompt += "<|im_start|>assistant\n"
|
||||||
# prompt += f"<|im_start|>assistant\n"
|
|
||||||
return prompt
|
return prompt
|
||||||
|
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue