diff options
Diffstat (limited to 'modules')
-rw-r--r-- | modules/textual_inversion/dataset.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/modules/textual_inversion/dataset.py b/modules/textual_inversion/dataset.py index bd99c0cb..23bb4b6a 100644 --- a/modules/textual_inversion/dataset.py +++ b/modules/textual_inversion/dataset.py @@ -83,6 +83,7 @@ class PersonalizedBase(Dataset): self.dataset.append(entry)
+ assert len(self.dataset) > 1, "No images have been found in the dataset."
self.length = len(self.dataset) * repeats // batch_size
self.initial_indexes = np.arange(len(self.dataset))
|