Update forge_svd.py

This commit is contained in:
lllyasviel 2024-01-25 22:03:43 -08:00
parent a0694c856b
commit 447b40cc5e

View File

@ -4,6 +4,7 @@ import pathlib
from modules import scripts, script_callbacks from modules import scripts, script_callbacks
from modules.paths import models_path from modules.paths import models_path
from modules.ui_common import ToolButton, refresh_symbol
from modules import shared from modules import shared
@ -36,8 +37,14 @@ def on_ui_tabs():
with gr.Blocks(analytics_enabled=False) as svd_block: with gr.Blocks(analytics_enabled=False) as svd_block:
with gr.Row(): with gr.Row():
with gr.Column(): with gr.Column():
width = gr.Slider(label="width", minimum=64, maximum=2048, value=512, step=64, interactive=True) with gr.Row():
height = gr.Slider(label="height", minimum=64, maximum=2048, value=512, step=64, interactive=True) width = gr.Slider(label="width", minimum=64, maximum=2048, value=512, step=64, interactive=True)
refresh_button = ToolButton(value=refresh_symbol, tooltip="Refresh")
# refresh_button.click(
# fn=refresh,
# inputs=[],
# outputs=refresh_components
# )
with gr.Column(): with gr.Column():
png_output = gr.Button(value="Save PNG") png_output = gr.Button(value="Save PNG")