|
|
|
|
@ -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()]),
|
|
|
|
|
|