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

Merge pull request #1205 from feffy380/patch-1

Browse Source 
register reg images with correct subset
This commit is contained in:
Kohya S
2024-03-24 11:14:07 +09:00
committed by GitHub
parent 691f04322a 0c7baea88c
commit ad97410ba5
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