update chatglm_llm.py
This commit is contained in:
parent
bb8224025e
commit
46b872a854
|
|
@ -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
|
||||||
)):
|
)):
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue