diff --git a/modules/ui.py b/modules/ui.py
index 179e3a83e..22ca74c2c 100644
--- a/modules/ui.py
+++ b/modules/ui.py
@@ -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()]),