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

update new locon args support

Browse Source
This commit is contained in:
Ki-wimon 2023-03-01 12:22:11 +08:00 committed by GitHub
parent c07e3bba76
commit 6bcd52c9cc
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
lora_gui.py
View File

@ -116,7 +116,8 @@ def save_configuration(
bucket_reso_steps, bucket_reso_steps,
caption_dropout_every_n_epochs, caption_dropout_rate, caption_dropout_every_n_epochs, caption_dropout_rate,
optimizer, optimizer,
optimizer_args,noise_offset, locon = 0 optimizer_args,noise_offset,
locon=0, conv_dim=0, conv_alpha=0,
): ):
# Get list of function parameters and values # Get list of function parameters and values
parameters = list(locals().items()) parameters = list(locals().items())
@ -222,7 +223,8 @@ def open_configuration(
bucket_reso_steps, bucket_reso_steps,
caption_dropout_every_n_epochs, caption_dropout_rate, caption_dropout_every_n_epochs, caption_dropout_rate,
optimizer, optimizer,
optimizer_args,noise_offset, locon=0 optimizer_args,noise_offset,
locon=0, conv_dim=0, conv_alpha=0,
): ):
# Get list of function parameters and values # Get list of function parameters and values
parameters = list(locals().items()) parameters = list(locals().items())
@ -306,7 +308,8 @@ def train_model(
bucket_reso_steps, bucket_reso_steps,
caption_dropout_every_n_epochs, caption_dropout_rate, caption_dropout_every_n_epochs, caption_dropout_rate,
optimizer, optimizer,
optimizer_args,noise_offset, locon optimizer_args,noise_offset,
locon, conv_dim, conv_alpha,
): ):
if pretrained_model_name_or_path == '': if pretrained_model_name_or_path == '':
msgbox('Source model information is missing') msgbox('Source model information is missing')
@ -852,7 +855,8 @@ def lora_tab(
bucket_reso_steps, bucket_reso_steps,
caption_dropout_every_n_epochs, caption_dropout_rate, caption_dropout_every_n_epochs, caption_dropout_rate,
optimizer, optimizer,
optimizer_args,noise_offset,locon optimizer_args,noise_offset,
locon, conv_dim, conv_alpha,
] ]
button_open_config.click( button_open_config.click(