diff --git a/app/models/qwen.py b/app/models/qwen.py index e340728..24c2bfa 100644 --- a/app/models/qwen.py +++ b/app/models/qwen.py @@ -67,8 +67,7 @@ class Qwen: prompt += f"<|im_start|>system\n{message['text']}<|im_end|>\n\n" elif message["role"] == "user": prompt += f"<|im_start|>user\n{message['text']}<|im_end|>\n\n" - #elif message["role"] == "assistant": - # prompt += f"<|im_start|>assistant\n" + prompt += "<|im_start|>assistant\n" return prompt