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

register reg images with correct subset

Browse Source
This commit is contained in:
feffy380
2024-03-23 17:28:02 +01:00
committed by GitHub
parent f4a4c11cd3
commit 0c7baea88c
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
library/train_util.py
View File

@@ -1554,7 +1554,7 @@ class DreamBoothDataset(BaseDataset):
for img_path, caption in zip(img_paths, captions): for img_path, caption in zip(img_paths, captions):
info = ImageInfo(img_path, subset.num_repeats, caption, subset.is_reg, img_path) info = ImageInfo(img_path, subset.num_repeats, caption, subset.is_reg, img_path)
if subset.is_reg: if subset.is_reg:
reg_infos.append(info) reg_infos.append((info, subset))
else: else:
self.register_image(info, subset) self.register_image(info, subset)
@@ -1575,7 +1575,7 @@ class DreamBoothDataset(BaseDataset):
n = 0 n = 0
first_loop = True first_loop = True
while n < num_train_images: while n < num_train_images:
for info in reg_infos: for info, subset in reg_infos:
if first_loop: if first_loop:
self.register_image(info, subset) self.register_image(info, subset)
n += info.num_repeats n += info.num_repeats