From d7e259cedf4cfaa7c968d95fbc7d3811bc9cf6ba Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Thu, 25 Jan 2024 22:10:23 -0800 Subject: [PATCH] Update forge_svd.py --- extensions-builtin/sd_forge_svd/scripts/forge_svd.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/extensions-builtin/sd_forge_svd/scripts/forge_svd.py b/extensions-builtin/sd_forge_svd/scripts/forge_svd.py index a6606de4..18f75d8b 100644 --- a/extensions-builtin/sd_forge_svd/scripts/forge_svd.py +++ b/extensions-builtin/sd_forge_svd/scripts/forge_svd.py @@ -45,9 +45,11 @@ def on_ui_tabs(): refresh_button.click( fn=lambda: gr.update(choices=update_svd_filenames), inputs=[], outputs=filename) + generate_button = gr.Button(value="Generate") with gr.Column(): - png_output = gr.Button(value="Save PNG") + output_gallery = gr.Gallery(label='Gallery', show_label=False, object_fit='contain', + visible=True, height=1024, columns=4) return [(svd_block, "SVD", "svd")]