From 9136fc201538d917024770682b0e64f3cb71fee2 Mon Sep 17 00:00:00 2001
From: Oleg Zakharov <ozakharov@gpu-2-2-19.int.aldan3.itm-rsmu.ru>
Date: Thu, 17 Oct 2024 02:03:58 +0300
Subject: [PATCH] v2

---
 app/models/qwen.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

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