ros/LLaMA-Factory
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

back to gradio 4.21 and fix chat

Browse Source 
Former-commit-id: 695734a40a702ea059d855da54080cc8d161e41a
This commit is contained in:
hiyouga
2024-04-04 02:07:20 +08:00
parent b1986a06b9
commit 48ceac845c
5 changed files with 27 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
src/llmtuner/webui/components/chatbot.py
View File

@@ -35,13 +35,15 @@ def create_chat_box(
tools.input(check_json_schema, inputs=[tools, engine.manager.get_elem_by_id("top.lang")])
submit_btn.click(
engine.chatter.predict,
[chatbot, role, query, messages, system, tools, max_new_tokens, top_p, temperature],
engine.chatter.append,
[chatbot, messages, role, query],
[chatbot, messages, query],
).then(
engine.chatter.stream,
[chatbot, messages, system, tools, max_new_tokens, top_p, temperature],
[chatbot, messages],
show_progress=True,
).then(lambda: "", outputs=[query])
clear_btn.click(lambda: ([], []), outputs=[chatbot, messages], show_progress=True)
)
clear_btn.click(lambda: ([], []), outputs=[chatbot, messages])
return (
chat_box,