Fix standalone lora gui issue

2022-12-30 21:18:50 -05:00 · 2022-12-30 21:18:50 -05:00 · 6b41f4f951
commit 6b41f4f951
parent 4216d05471
1 changed files with 2 additions and 2 deletions
--- a/lora_gui.py
+++ b/lora_gui.py
@ -503,13 +503,13 @@ def UI(username, password):
    interface = gr.Blocks(css=css)

    with interface:
-        with gr.Tab('Dreambooth'):
+        with gr.Tab('LoRA'):
            (
                train_data_dir_input,
                reg_data_dir_input,
                output_dir_input,
                logging_dir_input,
-            ) = dreambooth_tab()
+            ) = lora_tab()
        with gr.Tab('Utilities'):
            utilities_tab(
                train_data_dir_input=train_data_dir_input,