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

Add network_args to meta if using another network

Browse Source
This commit is contained in:
Isotr0py
2023-03-04 13:59:22 +08:00
parent 45945f698a
commit 4396350271
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