Fix stop encoder training issue

bmaltais · Dec 19, 2022 · 6987f51 · 6987f51
1 parent 61a9481
commit 6987f51
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 11 deletions.
diff --git a/README.md b/README.md
@@ -130,6 +130,8 @@ Drop by the discord server for support: https://discord.com/channels/10415185624
 
 ## Change history
 
+* 12/19 (v18.3) update:
+    - fix stop encoder training issue
 * 12/19 (v18.2) update:
     - Fix file/folder opening behind the browser window
     - Add WD14 and BLIP captioning to utilities

diff --git a/dreambooth_gui.py b/dreambooth_gui.py
@@ -381,7 +381,8 @@ def save_inference_file(output_dir, v2, v_parameterization):
     run_cmd += f' --save_precision={save_precision}'
     run_cmd += f' --logging_dir={logging_dir}'
     run_cmd += f' --caption_extention={caption_extention}'
-    run_cmd += f' --stop_text_encoder_training={stop_text_encoder_training}'
+    if not stop_text_encoder_training == 0:
+        run_cmd += f' --stop_text_encoder_training={stop_text_encoder_training}'
     if not save_model_as == 'same as source model':
         run_cmd += f' --save_model_as={save_model_as}'
 
@@ -462,22 +463,23 @@ def set_pretrained_model_name_or_path_input(value, v2, v_parameterization):
     dummy_false = gr.Label(value=False, visible=False)
     with gr.Tab('Dreambooth'):
         gr.Markdown('Enter kohya finetuner parameter using this interface.')
-        with gr.Accordion('Configuration File Load/Save', open=False):
+        with gr.Accordion('Configuration file', open=False):
             with gr.Row():
                 button_open_config = gr.Button('Open 📂', elem_id='open_folder')
                 button_save_config = gr.Button('Save 💾', elem_id='open_folder')
                 button_save_as_config = gr.Button(
                     'Save as... 💾', elem_id='open_folder'
                 )
-            config_file_name = gr.Textbox(
-                label='',
-                placeholder="type the configuration file path or use the 'Open' button above to select it...",
-            )
-            config_file_name.change(
-                remove_doublequote,
-                inputs=[config_file_name],
-                outputs=[config_file_name],
-            )
+                config_file_name = gr.Textbox(
+                    label='',
+                    # placeholder="type the configuration file path or use the 'Open' button above to select it...",
+                    interactive=False
+                )
+            # config_file_name.change(
+            #     remove_doublequote,
+            #     inputs=[config_file_name],
+            #     outputs=[config_file_name],
+            # )
         with gr.Tab('Source model'):
             # Define the input elements
             with gr.Row():