Add option to not train TE in Dreambooth
This commit is contained in:
parent
25d84ecffe
commit
87622238c6
@ -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
|
||||
|
@ -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,
|
||||
|
Loading…
Reference in New Issue
Block a user