From 25b5caced80d105e865de465e11a97545ebae47b Mon Sep 17 00:00:00 2001 From: DKnight54 <126916963+DKnight54@users.noreply.github.com> Date: Mon, 26 May 2025 02:03:15 +0800 Subject: [PATCH] Update train_util.py --- library/train_util.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/library/train_util.py b/library/train_util.py index 10b24da2..b958e189 100644 --- a/library/train_util.py +++ b/library/train_util.py @@ -1765,11 +1765,7 @@ class DreamBoothDataset(BaseDataset): continue if subset.is_reg: - if subset.num_repeats > 1: - info.num_repeats = 1 - self.reg_infos[info.image_key] = (info, subset) - for i in range(subset.num_repeats): - self.reg_infos_index.append(info.image_key) + num_reg_images += subset.num_repeats * len(img_paths) else: num_train_images += subset.num_repeats * len(img_paths) @@ -1778,7 +1774,11 @@ 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 + self.reg_infos[info.image_key] = (info, subset) + for i in range(subset.num_repeats): + self.reg_infos_index.append(info.image_key) else: self.register_image(info, subset)