Merge branch 'kohya-ss:main' into flux_shift

This commit is contained in:
青龍聖者@bdsqlsz
2024-09-05 13:54:15 +08:00
committed by GitHub
2 changed files with 2 additions and 2 deletions

View File

@@ -1708,7 +1708,7 @@ class DreamBoothDataset(BaseDataset):
def load_dreambooth_dir(subset: DreamBoothSubset):
if not os.path.isdir(subset.image_dir):
logger.warning(f"not directory: {subset.image_dir}")
return [], []
return [], [], []
info_cache_file = os.path.join(subset.image_dir, self.IMAGE_INFO_CACHE_FILE)
use_cached_info_for_subset = subset.cache_info