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

fixed generate_controlnet_subsets_config for training

Browse Source
This commit is contained in:
青龍聖者@bdsqlsz
2023-09-16 18:12:21 +08:00
parent 55cdd5c9ae
commit ae606ace74
1 changed files with 3 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
library/config_util.py
View File

@@ -547,17 +547,13 @@ def generate_controlnet_subsets_config_by_subdirs(train_data_dir: Optional[str]
return []
subsets_config = []
for subdir in base_dir.iterdir():
if not subdir.is_dir():
continue
subset_config = {"image_dir": str(subdir), "conditioning_data_dir": conditioning_data_dir, "caption_extension": caption_extension, "num_repeats": 1}
subsets_config.append(subset_config)
subset_config = {"image_dir": train_data_dir, "conditioning_data_dir": conditioning_data_dir, "caption_extension": caption_extension, "num_repeats": 1}
subsets_config.append(subset_config)
return subsets_config
subsets_config = []
subsets_config += generate(train_data_dir, False)
subsets_config += generate(train_data_dir)
return subsets_config