Add option to not train TE in Dreambooth

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

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):
- 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):
- Fix issue with kohya locon not training the convolution layers
- Update LyCORIS module version

View File

@ -377,7 +377,9 @@ def train_model(
print(f'max_train_steps = {max_train_steps}')
# 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
else:
stop_text_encoder_training = math.ceil(
@ -624,7 +626,7 @@ def dreambooth_tab(
placeholder='512,512',
)
stop_text_encoder_training = gr.Slider(
minimum=0,
minimum=-1,
maximum=100,
value=0,
step=1,