Strip sdxl-turbo options

nod-ai · Jun 17, 2024 · 84bc143 · 84bc143
1 parent 83f424a
commit 84bc143
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/apps/shark_studio/web/ui/sd.py b/apps/shark_studio/web/ui/sd.py
@@ -51,7 +51,7 @@
     # "stabilityai/stable-diffusion-2-1-base",
     # "stabilityai/stable-diffusion-2-1",
     # "stabilityai/stable-diffusion-xl-base-1.0",
-    "stabilityai/sdxl-turbo",
+    #"stabilityai/sdxl-turbo",
 ]
 sd_default_models.extend(get_checkpoints(model_type="scripts"))
 
@@ -402,6 +402,7 @@ def base_model_changed(base_model_id):
                             value=init_config["scheduler"],
                             choices=scheduler_model_map.keys(),
                             allow_custom_value=False,
+                            visible=False,
                         )
                     with gr.Row():
                         steps = gr.Dropdown(
@@ -416,6 +417,7 @@ def base_model_changed(base_model_id):
                             value=4,
                             step=0.1,
                             label="\U0001F5C3\U0000FE0F CFG Scale",
+                            visible=False,
                         )
                     with gr.Row():
                         batch_count = gr.Slider(
@@ -425,7 +427,7 @@ def base_model_changed(base_model_id):
                             step=1,
                             label="Batch Count",
                             interactive=True,
-                            visible=True,
+                            visible=False,
                         )
                         batch_size = gr.Slider(
                             1,
@@ -434,7 +436,7 @@ def base_model_changed(base_model_id):
                             step=1,
                             label="Batch Size",
                             interactive=False,  # DEMO
-                            visible=True,
+                            visible=False,
                         )
                         compiled_pipeline = gr.Checkbox(
                             value=init_config["compiled_pipeline"],
@@ -451,13 +453,14 @@ def base_model_changed(base_model_id):
                             elem_classes=["fill"],
                             value=view_json_file(default_config_file),
                         )
-                    with gr.Row():
+                    with gr.Row(visible=False):
                         with gr.Row():
                             load_sd_config = gr.Dropdown(
                                 label="Load Config",
                                 value=cmd_opts.defaults,
                                 choices=get_configs(),
                                 allow_custom_value=True,
+                                visible=False,
                             )
                         with gr.Row():
                             save_sd_config = gr.Button(