Commit 8ec069e6 authored by JC_Array's avatar JC_Array

removed duplicate run_preprocess.click by creating run_preprocess_inputs list...

removed duplicate run_preprocess.click by creating run_preprocess_inputs list and appending deepbooru variable to input list if in scope
parent 1f92336b
......@@ -1089,35 +1089,20 @@ def create_ui(wrap_gradio_gpu_call):
]
)
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=[
run_preprocess_inputs = [
process_src,
process_dst,
process_flip,
process_split,
process_caption,
process_caption_deepbooru,
],
outputs=[
ti_output,
ti_outcome,
],
)
else:
]
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_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=[
process_src,
process_dst,
process_flip,
process_split,
process_caption,
],
inputs=run_preprocess_inputs,
outputs=[
ti_output,
ti_outcome,
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment