From e38392b950df160d2c3217df52d95a23e8d02f51 Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Thu, 25 Jan 2024 22:07:07 -0800 Subject: [PATCH] Update forge_svd.py --- extensions-builtin/sd_forge_svd/scripts/forge_svd.py | 10 ++++------ 1 file changed, 4 insertions(+), 6 deletions(-) diff --git a/extensions-builtin/sd_forge_svd/scripts/forge_svd.py b/extensions-builtin/sd_forge_svd/scripts/forge_svd.py index f1885d55..b065ba23 100644 --- a/extensions-builtin/sd_forge_svd/scripts/forge_svd.py +++ b/extensions-builtin/sd_forge_svd/scripts/forge_svd.py @@ -38,13 +38,11 @@ def on_ui_tabs(): with gr.Row(): with gr.Column(): with gr.Row(): - width = gr.Slider(label="width", minimum=64, maximum=2048, value=512, step=64, interactive=True) + filename = gr.Dropdown(label="SVD Checkpoint Filename", choices=svd_filenames, value=svd_filenames[0] if len(svd_filenames) > 0 else None) refresh_button = ToolButton(value=refresh_symbol, tooltip="Refresh") - # refresh_button.click( - # fn=refresh, - # inputs=[], - # outputs=refresh_components - # ) + refresh_button.click( + fn=lambda: gr.update(choices=update_svd_filenames), + inputs=[], outputs=filename) with gr.Column(): png_output = gr.Button(value="Save PNG")