VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-10 15:00:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix handling default state

Browse Source
This commit is contained in:
rockerBOO
2023-11-23 10:00:01 -05:00
parent 4aba1217b0
commit aa04692f77
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
train_network.py
View File

@@ -504,7 +504,10 @@ class NetworkTrainer:
"ss_max_train_steps": args.max_train_steps, "ss_max_train_steps": args.max_train_steps,
"ss_lr_warmup_steps": args.lr_warmup_steps, "ss_lr_warmup_steps": args.lr_warmup_steps,
"ss_lr_scheduler": (args.lr_scheduler_type or args.lr_scheduler) "ss_lr_scheduler": (args.lr_scheduler_type or args.lr_scheduler)
+ (f"({args.lr_scheduler_args})" if len(args.lr_scheduler_args) > 0 else ""), + (f"({args.lr_scheduler_args})"
if args.lr_scheduler_args is not None
and len(args.lr_scheduler_args) > 0
else ""),
"ss_network_module": args.network_module, "ss_network_module": args.network_module,
"ss_network_dim": args.network_dim, # None means default because another network than LoRA may have another default dim "ss_network_dim": args.network_dim, # None means default because another network than LoRA may have another default dim
"ss_network_alpha": args.network_alpha, # some networks may not have alpha "ss_network_alpha": args.network_alpha, # some networks may not have alpha