提交 25e5ab12 作者: glide-the

增加call_行为

上级 c4ee36b8
...@@ -39,7 +39,14 @@ class ChatGLM(BaseAnswer, LLM, ABC): ...@@ -39,7 +39,14 @@ class ChatGLM(BaseAnswer, LLM, ABC):
self.history_len = history_len self.history_len = history_len
def _call(self, prompt: str, stop: Optional[List[str]] = None) -> str: def _call(self, prompt: str, stop: Optional[List[str]] = None) -> str:
pass response, _ = self.checkPoint.model.chat(
self.checkPoint.tokenizer,
prompt,
history=[],
max_length=self.max_token,
temperature=self.temperature
)
return response
def generatorAnswer(self, prompt: str, def generatorAnswer(self, prompt: str,
history: List[List[str]] = [], history: List[List[str]] = [],
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论