shaobo преди 1 година
родител
ревизия
feaa9d6b23
променени са 1 файла, в които са добавени 1 реда и са изтрити 1 реда
  1. 1 1
      local_mode/models/codegeex.py

+ 1 - 1
local_mode/models/codegeex.py

@@ -59,7 +59,7 @@ class CodegeexChatModel:
         try:
             response, _ = self.model.chat(
                 self.tokenizer,
-                query=request.messages[0].content,
+                query=request.messages[-1].content,
                 history=[msg.model_dump() for msg in request.messages[:-1]],
                 max_new_tokens=request.max_tokens,
                 temperature=request.temperature,