提交 46b872a8 作者: imClumsyPanda

update chatglm_llm.py

上级 bb822402
...@@ -52,7 +52,7 @@ class ChatGLM(BaseAnswer, LLM, ABC): ...@@ -52,7 +52,7 @@ class ChatGLM(BaseAnswer, LLM, ABC):
for inum, (stream_resp, _) in enumerate(self.checkPoint.model.stream_chat( for inum, (stream_resp, _) in enumerate(self.checkPoint.model.stream_chat(
self.checkPoint.tokenizer, self.checkPoint.tokenizer,
prompt, prompt,
history=history[-self.history_len:] if self.history_len > 0 else [], history=history[-self.history_len:-1] if self.history_len > 1 else [],
max_length=self.max_token, max_length=self.max_token,
temperature=self.temperature temperature=self.temperature
)): )):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论