StableDiffusion/KohyaSS
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix stop encoder training issue

Browse Source
This commit is contained in:
bmaltais 2022-12-19 10:39:04 -05:00
parent 61a9481903
commit 6987f51b0a
2 changed files with 15 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@ -130,6 +130,8 @@ Drop by the discord server for support: https://discord.com/channels/10415185624
## Change history ## Change history
* 12/19 (v18.3) update:
- fix stop encoder training issue
* 12/19 (v18.2) update: * 12/19 (v18.2) update:
- Fix file/folder opening behind the browser window - Fix file/folder opening behind the browser window
- Add WD14 and BLIP captioning to utilities - Add WD14 and BLIP captioning to utilities

24
dreambooth_gui.py
View File

@ -381,7 +381,8 @@ def train_model(
run_cmd += f' --save_precision={save_precision}' run_cmd += f' --save_precision={save_precision}'
run_cmd += f' --logging_dir={logging_dir}' run_cmd += f' --logging_dir={logging_dir}'
run_cmd += f' --caption_extention={caption_extention}' 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': if not save_model_as == 'same as source model':
run_cmd += f' --save_model_as={save_model_as}' run_cmd += f' --save_model_as={save_model_as}'
@ -462,22 +463,23 @@ with interface:
dummy_false = gr.Label(value=False, visible=False) dummy_false = gr.Label(value=False, visible=False)
with gr.Tab('Dreambooth'): with gr.Tab('Dreambooth'):
gr.Markdown('Enter kohya finetuner parameter using this interface.') 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(): with gr.Row():
button_open_config = gr.Button('Open 📂', elem_id='open_folder') button_open_config = gr.Button('Open 📂', elem_id='open_folder')
button_save_config = gr.Button('Save 💾', elem_id='open_folder') button_save_config = gr.Button('Save 💾', elem_id='open_folder')
button_save_as_config = gr.Button( button_save_as_config = gr.Button(
'Save as... 💾', elem_id='open_folder' 'Save as... 💾', elem_id='open_folder'
) )
config_file_name = gr.Textbox( config_file_name = gr.Textbox(
label='', label='',
placeholder="type the configuration file path or use the 'Open' button above to select it...", # placeholder="type the configuration file path or use the 'Open' button above to select it...",
) interactive=False
config_file_name.change( )
remove_doublequote, # config_file_name.change(
inputs=[config_file_name], # remove_doublequote,
outputs=[config_file_name], # inputs=[config_file_name],
) # outputs=[config_file_name],
# )
with gr.Tab('Source model'): with gr.Tab('Source model'):
# Define the input elements # Define the input elements
with gr.Row(): with gr.Row():