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

Merge aa04692f77 into 1dae34b0af

Browse Source
This commit is contained in:
Dave Lage
2026-03-31 14:37:53 +00:00
committed by GitHub
parent 1dae34b0af aa04692f77
commit b204795c61
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
train_network.py
View File

@@ -999,7 +999,11 @@ class NetworkTrainer:
"ss_gradient_accumulation_steps": args.gradient_accumulation_steps,
"ss_max_train_steps": args.max_train_steps,
"ss_lr_warmup_steps": args.lr_warmup_steps,
"ss_lr_scheduler": args.lr_scheduler,
"ss_lr_scheduler": (args.lr_scheduler_type or args.lr_scheduler)
+ (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_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