v2

2024-10-17 02:03:58 +03:00 · 2024-10-17 02:03:58 +03:00 · 9136fc2015
parent 0a0fd07d2e
commit 9136fc2015
1 changed files with 1 additions and 2 deletions
--- 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