VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-16 17:02:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update train_util.py

Browse Source
This commit is contained in:
DKnight54
2025-05-26 02:03:15 +08:00
committed by GitHub
parent e183bc85c7
commit 25b5caced8
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
library/train_util.py
View File

@@ -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)