34 lines
851 B
Python
34 lines
851 B
Python
|
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)
|