Fix to work with validation dataset

2026-04-08 22:35:09 +00:00 · 2025-01-26 21:21:21 +09:00
parent b833d47afe
commit 58b82a576e
2 changed files with 4 additions and 1 deletions
--- a/library/train_util.py
+++ b/library/train_util.py
@@ -2403,6 +2403,7 @@ class ControlNetDataset(BaseDataset):

        self.dreambooth_dataset_delegate = DreamBoothDataset(
            db_subsets,
+            True,
            batch_size,
            resolution,
            network_multiplier,