Merge branch 'main' into dev

2026-04-08 22:35:09 +00:00 · 2024-09-07 10:52:49 +09:00
parent 0005867ba5 de25945a93
commit 62ec3e6424
3 changed files with 3 additions and 3 deletions
--- a/library/train_util.py
+++ b/library/train_util.py
@@ -1554,7 +1554,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