removed duplicate run_preprocess.click by creating run_preprocess_inputs list and appending deepbooru variable to input list if in scope
This commit is contained in:
parent
1f92336be7
commit
8ec069e64d
|
@ -1089,35 +1089,20 @@ 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 cmd_opts.deepdanbooru:
|
||||||
# if process_caption_deepbooru is None, it will cause an error, as a result only include it if it is enabled
|
# 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(
|
run_preprocess.click(
|
||||||
fn=wrap_gradio_gpu_call(modules.textual_inversion.ui.preprocess, extra_outputs=[gr.update()]),
|
fn=wrap_gradio_gpu_call(modules.textual_inversion.ui.preprocess, extra_outputs=[gr.update()]),
|
||||||
_js="start_training_textual_inversion",
|
_js="start_training_textual_inversion",
|
||||||
inputs=[
|
inputs=run_preprocess_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=[
|
outputs=[
|
||||||
ti_output,
|
ti_output,
|
||||||
ti_outcome,
|
ti_outcome,
|
||||||
|
|
Loading…
Reference in New Issue