diff options
| author | JC_Array <justin318@att.net> | 2022-10-10 03:23:24 -0500 |
|---|---|---|
| committer | JC_Array <justin318@att.net> | 2022-10-10 03:23:24 -0500 |
| commit | 8ec069e64df48f8f202f8b93a08e91b69448eb39 (patch) | |
| tree | 7d697a8c04870b62d98270b929e07d54be5bc6a5 /modules/ui.py | |
| parent | 1f92336be768d235c18a82acb2195b7135101ae7 (diff) | |
removed duplicate run_preprocess.click by creating run_preprocess_inputs list and appending deepbooru variable to input list if in scope
Diffstat (limited to 'modules/ui.py')
| -rw-r--r-- | modules/ui.py | 49 |
1 files changed, 17 insertions, 32 deletions
diff --git a/modules/ui.py b/modules/ui.py index 179e3a83..22ca74c2 100644 --- a/modules/ui.py +++ b/modules/ui.py @@ -1089,40 +1089,25 @@ def create_ui(wrap_gradio_gpu_call): ]
)
+ run_preprocess_inputs = [
+ process_src,
+ process_dst,
+ process_flip,
+ process_split,
+ process_caption,
+ ]
if cmd_opts.deepdanbooru:
# if process_caption_deepbooru is None, it will cause an error, as a result only include it if it is enabled
- run_preprocess.click(
- fn=wrap_gradio_gpu_call(modules.textual_inversion.ui.preprocess, extra_outputs=[gr.update()]),
- _js="start_training_textual_inversion",
- inputs=[
- process_src,
- process_dst,
- process_flip,
- process_split,
- process_caption,
- process_caption_deepbooru,
- ],
- outputs=[
- ti_output,
- ti_outcome,
- ],
- )
- else:
- run_preprocess.click(
- fn=wrap_gradio_gpu_call(modules.textual_inversion.ui.preprocess, extra_outputs=[gr.update()]),
- _js="start_training_textual_inversion",
- inputs=[
- process_src,
- process_dst,
- process_flip,
- process_split,
- process_caption,
- ],
- outputs=[
- ti_output,
- ti_outcome,
- ],
- )
+ run_preprocess_inputs.append(process_caption_deepbooru)
+ run_preprocess.click(
+ fn=wrap_gradio_gpu_call(modules.textual_inversion.ui.preprocess, extra_outputs=[gr.update()]),
+ _js="start_training_textual_inversion",
+ inputs=run_preprocess_inputs,
+ outputs=[
+ ti_output,
+ ti_outcome,
+ ],
+ )
train_embedding.click(
fn=wrap_gradio_gpu_call(modules.textual_inversion.ui.train_embedding, extra_outputs=[gr.update()]),
|