Commit 2f94331d authored by JC_Array's avatar JC_Array

removed change in last commit, simplified to adding the visible argument to...

removed change in last commit, simplified to adding the visible argument to process_caption_deepbooru and it set to False if deepdanbooru argument is not set
parent 8ec069e6
...@@ -1036,7 +1036,8 @@ def create_ui(wrap_gradio_gpu_call): ...@@ -1036,7 +1036,8 @@ def create_ui(wrap_gradio_gpu_call):
process_caption = gr.Checkbox(label='Use BLIP caption as filename') process_caption = gr.Checkbox(label='Use BLIP caption as filename')
if cmd_opts.deepdanbooru: if cmd_opts.deepdanbooru:
process_caption_deepbooru = gr.Checkbox(label='Use deepbooru caption as filename') process_caption_deepbooru = gr.Checkbox(label='Use deepbooru caption as filename')
else:
process_caption_deepbooru = gr.Checkbox(label='Use deepbooru caption as filename', visible=False)
with gr.Row(): with gr.Row():
with gr.Column(scale=3): with gr.Column(scale=3):
...@@ -1089,20 +1090,17 @@ def create_ui(wrap_gradio_gpu_call): ...@@ -1089,20 +1090,17 @@ def create_ui(wrap_gradio_gpu_call):
] ]
) )
run_preprocess_inputs = [ 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_src,
process_dst, process_dst,
process_flip, process_flip,
process_split, process_split,
process_caption, process_caption,
] process_caption_deepbooru
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=run_preprocess_inputs,
outputs=[ outputs=[
ti_output, ti_output,
ti_outcome, 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