Create forge_svd.py
This commit is contained in:
parent
63f5f8e3f3
commit
6471999e5a
33
extensions-builtin/sd_forge_svd/scripts/forge_svd.py
Normal file
33
extensions-builtin/sd_forge_svd/scripts/forge_svd.py
Normal file
@ -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)
|
Loading…
Reference in New Issue
Block a user