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

fix num_repeats not working in DB classic dataset

Browse Source
This commit is contained in:
Kohya S
2023-03-02 19:01:22 +09:00
parent e9f37c4049
commit 83bfb54f20
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
library/config_util.py
View File

@@ -454,7 +454,7 @@ def generate_dreambooth_subsets_config_by_subdirs(train_data_dir: Optional[str]
if num_repeats < 1:
continue
subset_config = {"image_dir": str(subdir), "is_reg": is_reg, "class_tokens": class_tokens}
subset_config = {"image_dir": str(subdir), "num_repeats": num_repeats, "is_reg": is_reg, "class_tokens": class_tokens}
subsets_config.append(subset_config)
return subsets_config