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

Clear sizes for validation reg images to be consistent

Browse Source
This commit is contained in:
rockerBOO
2025-02-16 22:28:34 -05:00
parent 3c7496ae3f
commit f3a010978c
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
library/train_util.py
View File

@@ -1990,6 +1990,7 @@ class DreamBoothDataset(BaseDataset):
# Skip any validation dataset for regularization images # Skip any validation dataset for regularization images
if self.is_training_dataset is False: if self.is_training_dataset is False:
img_paths = [] img_paths = []
sizes = []
# Otherwise the img_paths remain as original img_paths and no split # Otherwise the img_paths remain as original img_paths and no split
# required for training images dataset of regularization images # required for training images dataset of regularization images
else: else: