add minimal impl for masked loss

2026-04-08 22:35:09 +00:00 · 2024-02-26 23:19:58 +09:00
parent 577e9913ca
commit f2c727fc8c
3 changed files with 45 additions and 14 deletions
--- a/library/train_util.py
+++ b/library/train_util.py
@@ -1810,6 +1810,9 @@ class ControlNetDataset(BaseDataset):

        db_subsets = []
        for subset in subsets:
+            assert (
+                not subset.random_crop
+            ), "random_crop is not supported in ControlNetDataset / random_cropはControlNetDatasetではサポートされていません"
            db_subset = DreamBoothSubset(
                subset.image_dir,
                False,