From 6987f51b0a7bbedc1d0770162ed458428cc58cdf Mon Sep 17 00:00:00 2001
From: bmaltais <bernard@ducourier.com>
Date: Mon, 19 Dec 2022 10:39:04 -0500
Subject: [PATCH] Fix stop encoder training issue

---
 README.md         |  2 ++
 dreambooth_gui.py | 24 +++++++++++++-----------
 2 files changed, 15 insertions(+), 11 deletions(-)

diff --git a/README.md b/README.md
index 5eec22ed4..a5d7c7718 100644
--- 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
index 0abe59f01..e8f370817 100644
--- 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():