This commit is contained in:
parent
c2b7ec20b8
commit
871d5994af
|
|
@ -112,7 +112,9 @@ with gr.Blocks(title="调试界面") as demo:
|
||||||
# Chatbot
|
# Chatbot
|
||||||
chatbot = gr.ChatInterface(
|
chatbot = gr.ChatInterface(
|
||||||
fn=chat,
|
fn=chat,
|
||||||
additional_inputs=[max_new, temp, top_p, top_k, rep_pen, pres_pen, stop_txt, log_state],
|
additional_inputs=[max_new, temp, top_p, top_k,
|
||||||
|
rep_pen, pres_pen, stop_txt, log_state],
|
||||||
|
additional_outputs=[log_state], # ★ 必加
|
||||||
type="messages"
|
type="messages"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue