Update chat_webui.py
Browse files- chat_webui.py +1 -1
chat_webui.py
CHANGED
@@ -85,7 +85,7 @@ with gr.Blocks(css=css) as demo:
|
|
85 |
max_length = gr.Slider(0, 4096, value=2048, step=1, label="Context Length", interactive=True)
|
86 |
top_p = gr.Slider(0, 1, value=0.7, step=0.01, label="Top-P", interactive=True)
|
87 |
temperature = gr.Slider(0, 1, value=0.5, step=0.01, label="Temperature", interactive=True)
|
88 |
-
rep_penalty = gr.Slider(1.0, 1.5, value=1.
|
89 |
|
90 |
|
91 |
submitBtn.click(predict, [user_input, chatbot, max_length, top_p, temperature, rep_penalty, gr.State(False)],
|
|
|
85 |
max_length = gr.Slider(0, 4096, value=2048, step=1, label="Context Length", interactive=True)
|
86 |
top_p = gr.Slider(0, 1, value=0.7, step=0.01, label="Top-P", interactive=True)
|
87 |
temperature = gr.Slider(0, 1, value=0.5, step=0.01, label="Temperature", interactive=True)
|
88 |
+
rep_penalty = gr.Slider(1.0, 1.5, value=1.1, step=0.01, label='Repetition Penalty', interactive=True)
|
89 |
|
90 |
|
91 |
submitBtn.click(predict, [user_input, chatbot, max_length, top_p, temperature, rep_penalty, gr.State(False)],
|