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

Add metadata

Browse Source
This commit is contained in:
rockerBOO
2025-02-14 14:18:53 -05:00
parent d0128d18be
commit 7729c4c8f9
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
train_network.py
View File

@@ -973,6 +973,7 @@ class NetworkTrainer:
"ss_max_validation_steps": args.max_validation_steps, "ss_max_validation_steps": args.max_validation_steps,
"ss_validate_every_n_epochs": args.validate_every_n_epochs, "ss_validate_every_n_epochs": args.validate_every_n_epochs,
"ss_validate_every_n_steps": args.validate_every_n_steps, "ss_validate_every_n_steps": args.validate_every_n_steps,
"ss_resize_interpolation": train_dataset_group.resize_interpolation
} }
self.update_metadata(metadata, args) # architecture specific metadata self.update_metadata(metadata, args) # architecture specific metadata
@@ -998,6 +999,7 @@ class NetworkTrainer:
"max_bucket_reso": dataset.max_bucket_reso, "max_bucket_reso": dataset.max_bucket_reso,
"tag_frequency": dataset.tag_frequency, "tag_frequency": dataset.tag_frequency,
"bucket_info": dataset.bucket_info, "bucket_info": dataset.bucket_info,
"resize_interpolation": dataset.resize_interpolation,
} }
subsets_metadata = [] subsets_metadata = []
@@ -1015,6 +1017,7 @@ class NetworkTrainer:
"enable_wildcard": bool(subset.enable_wildcard), "enable_wildcard": bool(subset.enable_wildcard),
"caption_prefix": subset.caption_prefix, "caption_prefix": subset.caption_prefix,
"caption_suffix": subset.caption_suffix, "caption_suffix": subset.caption_suffix,
"resize_interpolation": subset.resize_interpolation,
} }
image_dir_or_metadata_file = None image_dir_or_metadata_file = None