Create forge_svd.py

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)