Skip to content

Commit 6987f51

Browse files
committed
Fix stop encoder training issue
1 parent 61a9481 commit 6987f51

File tree

2 files changed

+15
-11
lines changed

2 files changed

+15
-11
lines changed

README.md

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -130,6 +130,8 @@ Drop by the discord server for support: https://discord.com/channels/10415185624
130130

131131
## Change history
132132

133+
* 12/19 (v18.3) update:
134+
- fix stop encoder training issue
133135
* 12/19 (v18.2) update:
134136
- Fix file/folder opening behind the browser window
135137
- Add WD14 and BLIP captioning to utilities

dreambooth_gui.py

Lines changed: 13 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -381,7 +381,8 @@ def save_inference_file(output_dir, v2, v_parameterization):
381381
run_cmd += f' --save_precision={save_precision}'
382382
run_cmd += f' --logging_dir={logging_dir}'
383383
run_cmd += f' --caption_extention={caption_extention}'
384-
run_cmd += f' --stop_text_encoder_training={stop_text_encoder_training}'
384+
if not stop_text_encoder_training == 0:
385+
run_cmd += f' --stop_text_encoder_training={stop_text_encoder_training}'
385386
if not save_model_as == 'same as source model':
386387
run_cmd += f' --save_model_as={save_model_as}'
387388

@@ -462,22 +463,23 @@ def set_pretrained_model_name_or_path_input(value, v2, v_parameterization):
462463
dummy_false = gr.Label(value=False, visible=False)
463464
with gr.Tab('Dreambooth'):
464465
gr.Markdown('Enter kohya finetuner parameter using this interface.')
465-
with gr.Accordion('Configuration File Load/Save', open=False):
466+
with gr.Accordion('Configuration file', open=False):
466467
with gr.Row():
467468
button_open_config = gr.Button('Open 📂', elem_id='open_folder')
468469
button_save_config = gr.Button('Save 💾', elem_id='open_folder')
469470
button_save_as_config = gr.Button(
470471
'Save as... 💾', elem_id='open_folder'
471472
)
472-
config_file_name = gr.Textbox(
473-
label='',
474-
placeholder="type the configuration file path or use the 'Open' button above to select it...",
475-
)
476-
config_file_name.change(
477-
remove_doublequote,
478-
inputs=[config_file_name],
479-
outputs=[config_file_name],
480-
)
473+
config_file_name = gr.Textbox(
474+
label='',
475+
# placeholder="type the configuration file path or use the 'Open' button above to select it...",
476+
interactive=False
477+
)
478+
# config_file_name.change(
479+
# remove_doublequote,
480+
# inputs=[config_file_name],
481+
# outputs=[config_file_name],
482+
# )
481483
with gr.Tab('Source model'):
482484
# Define the input elements
483485
with gr.Row():

0 commit comments

Comments
 (0)