接入新模型

This commit is contained in:
binary-husky
2023-10-28 19:23:43 +08:00
parent cf085565a7
commit 127385b846
18 changed files with 253 additions and 40 deletions

View File

@@ -219,7 +219,7 @@ def predict(inputs, llm_kwargs, plugin_kwargs, chatbot, history=[], system_promp
moss_handle = None
return
else:
response = "[Local Message]: 等待MOSS响应中 ..."
response = "[Local Message] 等待MOSS响应中 ..."
chatbot[-1] = (inputs, response)
yield from update_ui(chatbot=chatbot, history=history)
@@ -238,7 +238,7 @@ def predict(inputs, llm_kwargs, plugin_kwargs, chatbot, history=[], system_promp
yield from update_ui(chatbot=chatbot, history=history)
# 总结输出
if response == "[Local Message]: 等待MOSS响应中 ...":
response = "[Local Message]: MOSS响应异常 ..."
if response == "[Local Message] 等待MOSS响应中 ...":
response = "[Local Message] MOSS响应异常 ..."
history.extend([inputs, response.strip('<|MOSS|>: ')])
yield from update_ui(chatbot=chatbot, history=history)