update readme and typing hint

2026-04-09 06:45:09 +00:00 · 2024-03-24 11:27:18 +09:00
parent ad97410ba5
commit 381c44955e
2 changed files with 3 additions and 1 deletions
--- a/library/train_util.py
+++ b/library/train_util.py
@@ -1525,7 +1525,7 @@ class DreamBoothDataset(BaseDataset):
        logger.info("prepare images.")
        num_train_images = 0
        num_reg_images = 0
-        reg_infos: List[ImageInfo] = []
+        reg_infos: List[Tuple[ImageInfo, DreamBoothSubset]] = []
        for subset in subsets:
            if subset.num_repeats < 1:
                logger.warning(