incorrect arg order and support zero his

This commit is contained in:
lurenlym 2023-04-13 14:13:58 +08:00
parent bd39b87f67
commit a04244cf6a
2 changed files with 3 additions and 3 deletions

View File

@ -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,
) )

View File

@ -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="""
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
) )