Browse Source

Reorganize the sliders some more

oobabooga 3 years ago
parent
commit
0dd41e4830
2 changed files with 14 additions and 13 deletions
  1. 0 1
      modules/ui.py
  2. 14 12
      server.py

+ 0 - 1
modules/ui.py

@@ -26,7 +26,6 @@ css = """
   min-height: 0
 }
 #accordion {
-  background-color: transparent;
 }
 """
 

+ 14 - 12
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"):