hoelee/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-08-05 03:32:37 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Raise an assertion error if no training images have been found.

Browse Source
This commit is contained in:
Melan
2022-10-14 22:45:26 +02:00
parent 368f4cc4c7
commit 4d19f3b7d4
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
modules/textual_inversion/dataset.py
View File

@@ -82,6 +82,7 @@ class PersonalizedBase(Dataset):
self.dataset.append(entry) self.dataset.append(entry)
assert len(self.dataset) > 1, "No images have been found in the dataset."
self.length = len(self.dataset) * repeats self.length = len(self.dataset) * repeats
self.initial_indexes = np.arange(self.length) % len(self.dataset) self.initial_indexes = np.arange(self.length) % len(self.dataset)