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

do not save image_dir to metadata if None

Browse Source
This commit is contained in:
Kohya S
2023-03-01 23:37:17 +09:00
parent c95943b663
commit e9f37c4049
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
train_network.py
View File

@@ -323,7 +323,6 @@ def train(args):
subsets_metadata = []
for subset in dataset.subsets:
subset_metadata = {
"image_dir": os.path.basename(subset.image_dir),
"img_count": subset.img_count,
"num_repeats": subset.num_repeats,
"color_aug": bool(subset.color_aug),
@@ -332,6 +331,9 @@ def train(args):
"shuffle_caption": bool(subset.shuffle_caption),
"keep_tokens": subset.keep_tokens,
}
if subset.image_dir:
subset_metadata["image_dir"] = os.path.basename(subset.image_dir)
if is_dreambooth_dataset:
subset_metadata["class_tokens"] = subset.class_tokens
subset_metadata["is_reg"] = subset.is_reg