Update forge_svd.py

2026-02-01 05:49:45 +00:00 · 2024-01-25 22:10:23 -08:00
parent 67774fbd39
commit d7e259cedf
1 changed files with 3 additions and 1 deletions
--- 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")]