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

Add option to not train TE in Dreambooth

Browse Source
This commit is contained in:
bmaltais 2023-03-19 08:37:28 -04:00
parent 25d84ecffe
commit 87622238c6
2 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
README.md
View File

@ -191,6 +191,7 @@ This will store your a backup file with your current locally installed pip packa
* 2023/03/16 (v21.2.5): * 2023/03/16 (v21.2.5):
- Fix basic captioning logic - Fix basic captioning logic
- Add possibility to not train TE in Dreamboot by setting `Step text encoder training` to -1.
* 2023/03/12 (v21.2.4): * 2023/03/12 (v21.2.4):
- Fix issue with kohya locon not training the convolution layers - Fix issue with kohya locon not training the convolution layers
- Update LyCORIS module version - Update LyCORIS module version

6
dreambooth_gui.py
View File

@ -377,7 +377,9 @@ def train_model(
print(f'max_train_steps = {max_train_steps}') print(f'max_train_steps = {max_train_steps}')
# calculate stop encoder training # calculate stop encoder training
if stop_text_encoder_training_pct == None: if int(stop_text_encoder_training_pct) == -1:
stop_text_encoder_training = -1
elif stop_text_encoder_training_pct == None:
stop_text_encoder_training = 0 stop_text_encoder_training = 0
else: else:
stop_text_encoder_training = math.ceil( stop_text_encoder_training = math.ceil(
@ -624,7 +626,7 @@ def dreambooth_tab(
placeholder='512,512', placeholder='512,512',
) )
stop_text_encoder_training = gr.Slider( stop_text_encoder_training = gr.Slider(
minimum=0, minimum=-1,
maximum=100, maximum=100,
value=0, value=0,
step=1, step=1,