diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2022-10-15 10:13:23 +0300 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-10-15 10:13:23 +0300 |
| commit | 5207cba56f4f9c9a86c8ac40dc21bdf100b82625 (patch) | |
| tree | 864c371434923eb1b525e335ba71cab79f68a08e /modules/textual_inversion/dataset.py | |
| parent | a13af34b902bebc5df9509228380206a01f1245b (diff) | |
| parent | ea8aa1701a9f6ecb083356d345458f5baac852a5 (diff) | |
Merge pull request #2661 from Melanpan/master
Raise an assertion error if no training images have been found.
Diffstat (limited to 'modules/textual_inversion/dataset.py')
| -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))
|