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

Merge pull request #255 from Isotr0py/dev

Browse Source 
Add network_args to metadata if using another network
This commit is contained in:
Kohya S
2023-03-06 17:46:45 +09:00
committed by GitHub
parent 2b6e9d83fa 4396350271
commit c2a8290965
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
train_network.py
View File

@@ -427,9 +427,9 @@ def train(args):
"ss_bucket_info": json.dumps(dataset.bucket_info),
}
# uncomment if another network is added
# for key, value in net_kwargs.items():
# metadata["ss_arg_" + key] = value
if args.network_args:
for key, value in net_kwargs.items():
metadata["ss_arg_" + key] = value
if args.pretrained_model_name_or_path is not None:
sd_model_name = args.pretrained_model_name_or_path