From feaa9d6b2338493aebf1e250bb7628147e3a2ad4 Mon Sep 17 00:00:00 2001 From: shaobo Date: Thu, 11 Jul 2024 23:27:15 +0800 Subject: [PATCH] fix an index error --- local_mode/models/codegeex.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/local_mode/models/codegeex.py b/local_mode/models/codegeex.py index 8eda531..fd03fd5 100644 --- a/local_mode/models/codegeex.py +++ b/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,