Update train_util.py

2026-04-16 00:49:40 +00:00 · 2025-05-22 12:19:26 +08:00
parent 5e29081ee1
commit 0647284235
1 changed files with 4 additions and 1 deletions
--- a/library/train_util.py
+++ b/library/train_util.py
@@ -1721,7 +1721,10 @@ class DreamBoothDataset(BaseDataset):
                if size is not None:
                    info.image_size = size
                if subset.is_reg:
-                    reg_infos.append((info, subset))
+                    if subset.num_repeats > 1:
+                        info.num_repeats = 1
+                    for i in range(subset.num_repeats):
+                        reg_infos.append((info, subset))
                else:
                    self.register_image(info, subset)