Repetition penalty has to be < 5

This commit is contained in:
oobabooga 2023-02-07 23:23:39 -03:00
parent 1c30e1b49a
commit 7e4c25691d

View File

@ -387,7 +387,7 @@ def create_settings_menus():
top_p = gr.Slider(0.0,1.0,value=defaults['top_p'],step=0.01,label="top_p") top_p = gr.Slider(0.0,1.0,value=defaults['top_p'],step=0.01,label="top_p")
typical_p = gr.Slider(0.0,1.0,value=defaults['typical_p'],step=0.01,label="typical_p") typical_p = gr.Slider(0.0,1.0,value=defaults['typical_p'],step=0.01,label="typical_p")
with gr.Column(): with gr.Column():
repetition_penalty = gr.Slider(1.0,5.0,value=defaults['repetition_penalty'],step=0.01,label="repetition_penalty") repetition_penalty = gr.Slider(1.0,4.99,value=defaults['repetition_penalty'],step=0.01,label="repetition_penalty")
top_k = gr.Slider(0,200,value=defaults['top_k'],step=1,label="top_k") top_k = gr.Slider(0,200,value=defaults['top_k'],step=1,label="top_k")
no_repeat_ngram_size = gr.Slider(0, 20, step=1, value=defaults["no_repeat_ngram_size"], label="no_repeat_ngram_size") no_repeat_ngram_size = gr.Slider(0, 20, step=1, value=defaults["no_repeat_ngram_size"], label="no_repeat_ngram_size")