diff --git a/modules/ui.py b/modules/ui.py index b0fbc16..4aefc66 100644 --- a/modules/ui.py +++ b/modules/ui.py @@ -26,7 +26,6 @@ css = """ min-height: 0 } #accordion { - background-color: transparent; } """ diff --git a/server.py b/server.py index 84a4be0..7f4be43 100644 --- a/server.py +++ b/server.py @@ -462,23 +462,25 @@ def create_settings_menus(): with gr.Accordion("Custom generation parameters", open=False, elem_id="accordion"): with gr.Row(): - with gr.Column(): - do_sample = gr.Checkbox(value=generate_params['do_sample'], label="do_sample") - temperature = gr.Slider(0.01, 1.99, value=generate_params['temperature'], step=0.01, label="temperature") - top_p = gr.Slider(0.0,1.0,value=generate_params['top_p'],step=0.01,label="top_p") - typical_p = gr.Slider(0.0,1.0,value=generate_params['typical_p'],step=0.01,label="typical_p") - with gr.Column(): - top_k = gr.Slider(0,200,value=generate_params['top_k'],step=1,label="top_k") - repetition_penalty = gr.Slider(1.0,4.99,value=generate_params['repetition_penalty'],step=0.01,label="repetition_penalty") - no_repeat_ngram_size = gr.Slider(0, 20, step=1, value=generate_params["no_repeat_ngram_size"], label="no_repeat_ngram_size") - min_length = gr.Slider(0, 2000, step=1, value=generate_params["min_length"] if args.no_stream else 0, label="min_length", interactive=args.no_stream) + do_sample = gr.Checkbox(value=generate_params['do_sample'], label="do_sample") + temperature = gr.Slider(0.01, 1.99, value=generate_params['temperature'], step=0.01, label="temperature") + with gr.Row(): + top_k = gr.Slider(0,200,value=generate_params['top_k'],step=1,label="top_k") + top_p = gr.Slider(0.0,1.0,value=generate_params['top_p'],step=0.01,label="top_p") + with gr.Row(): + repetition_penalty = gr.Slider(1.0,4.99,value=generate_params['repetition_penalty'],step=0.01,label="repetition_penalty") + no_repeat_ngram_size = gr.Slider(0, 20, step=1, value=generate_params["no_repeat_ngram_size"], label="no_repeat_ngram_size") + with gr.Row(): + typical_p = gr.Slider(0.0,1.0,value=generate_params['typical_p'],step=0.01,label="typical_p") + min_length = gr.Slider(0, 2000, step=1, value=generate_params["min_length"] if args.no_stream else 0, label="min_length", interactive=args.no_stream) gr.Markdown("Contrastive search:") penalty_alpha = gr.Slider(0, 5, value=generate_params["penalty_alpha"], label="penalty_alpha") gr.Markdown("Beam search (uses a lot of VRAM):") - num_beams = gr.Slider(1, 20, step=1, value=generate_params["num_beams"], label="num_beams") - length_penalty = gr.Slider(-5, 5, value=generate_params["length_penalty"], label="length_penalty") + with gr.Row(): + num_beams = gr.Slider(1, 20, step=1, value=generate_params["num_beams"], label="num_beams") + length_penalty = gr.Slider(-5, 5, value=generate_params["length_penalty"], label="length_penalty") early_stopping = gr.Checkbox(value=generate_params["early_stopping"], label="early_stopping") with gr.Accordion("Soft prompt", open=False, elem_id="accordion"):