mirror of
https://github.com/kohya-ss/sd-scripts.git
synced 2026-04-06 21:52:27 +00:00
Merge pull request #261 from camenduru/main
metadata |= to metadata.update Thank you! I forget to fix this.
This commit is contained in:
@@ -408,7 +408,7 @@ def train(args):
|
||||
"img_count": subset.img_count
|
||||
}
|
||||
|
||||
metadata |= {
|
||||
metadata.update({
|
||||
"ss_batch_size_per_device": args.train_batch_size,
|
||||
"ss_total_batch_size": total_batch_size,
|
||||
"ss_resolution": args.resolution,
|
||||
@@ -425,7 +425,7 @@ def train(args):
|
||||
"ss_reg_dataset_dirs": json.dumps(reg_dataset_dirs_info),
|
||||
"ss_tag_frequency": json.dumps(dataset.tag_frequency),
|
||||
"ss_bucket_info": json.dumps(dataset.bucket_info),
|
||||
}
|
||||
})
|
||||
|
||||
# uncomment if another network is added
|
||||
# for key, value in net_kwargs.items():
|
||||
|
||||
Reference in New Issue
Block a user