From 84bc1437a715af9552991d0bf09a998783da850a Mon Sep 17 00:00:00 2001 From: Ean Garvey Date: Mon, 17 Jun 2024 17:57:40 -0500 Subject: [PATCH] Strip sdxl-turbo options --- apps/shark_studio/web/ui/sd.py | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/apps/shark_studio/web/ui/sd.py b/apps/shark_studio/web/ui/sd.py index 60fd487d9c..1f987c3e81 100644 --- 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(