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
This commit is contained in:
parent
8ec069e64d
commit
2f94331df2
|
@ -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):
|
||||||
]
|
]
|
||||||
)
|
)
|
||||||
|
|
||||||
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,
|
||||||
|
|
Loading…
Reference in New Issue