yang/CodeGeeX4
1
0
Fork 0
mirror of https://github.com/JasonYANG170/CodeGeeX4.git synced 2024-11-23 12:16:33 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix an index error

Browse Source
This commit is contained in:
shaobo 2024-07-11 23:27:15 +08:00
parent 0963521071
commit feaa9d6b23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
local_mode/models/codegeex.py
View File

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