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

1 files changed, 10 insertions, 12 deletions

diff --git a/modules/ui.py b/modules/ui.py
index 22ca74c2..f8adafb3 100644
--- a/modules/ui.py
+++ b/modules/ui.py
@@ -1036,7 +1036,8 @@ def create_ui(wrap_gradio_gpu_call):
                         process_caption = gr.Checkbox(label='Use BLIP caption as filename')

                         if cmd_opts.deepdanbooru:

                             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.Column(scale=3):

@@ -1089,20 +1090,17 @@ 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_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,

+            inputs=[

+                process_src,

+                process_dst,

+                process_flip,

+                process_split,

+                process_caption,

+                process_caption_deepbooru

+            ],

             outputs=[

                 ti_output,

                 ti_outcome,