From 6471999e5aafe40730ee590e687f8a546bbc07e9 Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Thu, 25 Jan 2024 21:48:59 -0800 Subject: [PATCH] Create forge_svd.py --- .../sd_forge_svd/scripts/forge_svd.py | 33 +++++++++++++++++++ 1 file changed, 33 insertions(+) create mode 100644 extensions-builtin/sd_forge_svd/scripts/forge_svd.py diff --git a/extensions-builtin/sd_forge_svd/scripts/forge_svd.py b/extensions-builtin/sd_forge_svd/scripts/forge_svd.py new file mode 100644 index 00000000..f494378c --- /dev/null +++ b/extensions-builtin/sd_forge_svd/scripts/forge_svd.py @@ -0,0 +1,33 @@ +import gradio as gr + +from modules import scripts, script_callbacks + + +class ForgeSVD(scripts.Script): + def __init__(self) -> None: + super().__init__() + + def title(self): + return "SVD" + + def show(self, is_img2img): + return scripts.AlwaysVisible + + def ui(self, is_img2img): + return () + + +def on_ui_tabs(): + with gr.Blocks(analytics_enabled=False) as svd_block: + with gr.Row(): + with gr.Column(): + width = gr.Slider(label="width", minimum=64, maximum=2048, value=512, step=64, interactive=True) + height = gr.Slider(label="height", minimum=64, maximum=2048, value=512, step=64, interactive=True) + + with gr.Column(): + png_output = gr.Button(value="Save PNG") + + return [(svd_block, "SVD", "svd")] + + +script_callbacks.on_ui_tabs(on_ui_tabs)