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