Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-20 01:31:13 +02:00
Code Issues Packages Projects Releases Wiki Activity

fully modularize ui.py

Browse Source
This commit is contained in:
Vladimir Mandic
2024-01-03 16:28:10 -05:00
parent f36a29ac68
commit 7ef6efd41c
13 changed files with 937 additions and 920 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/ui_train.py
+19 -2
View File
@@ -3,12 +3,13 @@ import gradio as gr
from modules import sd_hijack, script_callbacks, shared
from modules.ui_components import FormRow
from modules.ui_common import create_refresh_button
from modules.ui_sections import create_sampler_inputs
from modules.call_queue import wrap_gradio_gpu_call
from modules.textual_inversion import textual_inversion
import modules.errors
def create_ui(txt2img_preview_params):
def create_ui():
dummy_component = gr.Label(visible=False)
with gr.Row(elem_id="train_tab"):
@@ -41,8 +42,24 @@ def create_ui(txt2img_preview_params):
return gr_show(False), gr_show(True), gr_show(False)
elif tab == 'hn':
return gr_show(False), gr_show(False), gr_show(True)
else:
elif tab == 'pr':
return gr_show(True), gr_show(False), gr_show(False)
else:
return gr_show(False), gr_show(False), gr_show(False)
### preview tab
with gr.Tab(label="Preview settings", id="train_preview_tab") as tab_preview:
tab_preview.select(fn=lambda: train_tab_change('pr'), inputs=[], outputs=[action_pp, action_ti, action_hn])
prompt = gr.Textbox(label="Prompt", value="", placeholder="Prompt to be used for previews", lines=2)
negative = gr.Textbox(label="Negative prompt", value="", placeholder="Negative prompt to be used for previews", lines=2)
steps, sampler_index = create_sampler_inputs('train', accordion=False)
cfg_scale = gr.Slider(minimum=0.0, maximum=30.0, step=0.1, label='CFG scale', value=6.0)
seed = gr.Number(label='Initial seed', value=-1)
with gr.Row():
width = gr.Slider(minimum=64, maximum=8192, step=8, label="Width", value=512)
height = gr.Slider(minimum=64, maximum=8192, step=8, label="Height", value=512)
txt2img_preview_params = [prompt, negative, steps, sampler_index, cfg_scale, seed, width, height]
### preprocess tab