Unverified 提交 5c9e931a 作者: imClumsyPanda 提交者: GitHub

Merge pull request #73 from lurenlym/dev

incorrect arg order and support zero his
...@@ -38,7 +38,7 @@ class ChatGLM(LLM): ...@@ -38,7 +38,7 @@ class ChatGLM(LLM):
response, _ = self.model.chat( response, _ = self.model.chat(
self.tokenizer, self.tokenizer,
prompt, prompt,
history=self.history[-self.history_len:], history=self.history[-self.history_len:] if self.history_len>0 else [],
max_length=self.max_token, max_length=self.max_token,
temperature=self.temperature, temperature=self.temperature,
) )
......
...@@ -74,7 +74,7 @@ with gr.Blocks(css=""" ...@@ -74,7 +74,7 @@ with gr.Blocks(css="""
label="llm model", label="llm model",
value="chatglm-6b", value="chatglm-6b",
interactive=True) interactive=True)
LLM_HISTORY_LEN = gr.Slider(1, LLM_HISTORY_LEN = gr.Slider(0,
10, 10,
value=3, value=3,
step=1, step=1,
...@@ -95,7 +95,7 @@ with gr.Blocks(css=""" ...@@ -95,7 +95,7 @@ with gr.Blocks(css="""
kb.init_cfg(args[0], args[1], args[2], args[3]), kb.init_cfg(args[0], args[1], args[2], args[3]),
show_progress=True, show_progress=True,
api_name="init_cfg", api_name="init_cfg",
inputs=[llm_model, embedding_model, VECTOR_SEARCH_TOP_K, LLM_HISTORY_LEN] inputs=[llm_model, embedding_model, LLM_HISTORY_LEN,VECTOR_SEARCH_TOP_K]
).then( ).then(
get_model_status, chatbot, chatbot get_model_status, chatbot, chatbot
) )
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论