diff options
| author | Melan <alexleander91@gmail.com> | 2022-10-14 20:45:26 +0000 |
|---|---|---|
| committer | Melan <alexleander91@gmail.com> | 2022-10-14 20:45:26 +0000 |
| commit | 4d19f3b7d461fe0f63e7ccff936909b0ce0c6126 (patch) | |
| tree | c8bd49c107794fb3724d2a5c4e30f2774a4ebf03 | |
| parent | 368f4cc4c73509c1968cd9defe068d8bf4ff7c4f (diff) | |
| download | stable-diffusion-webui-gfx803-4d19f3b7d461fe0f63e7ccff936909b0ce0c6126.tar.gz stable-diffusion-webui-gfx803-4d19f3b7d461fe0f63e7ccff936909b0ce0c6126.tar.bz2 stable-diffusion-webui-gfx803-4d19f3b7d461fe0f63e7ccff936909b0ce0c6126.zip | |
Raise an assertion error if no training images have been found.
| -rw-r--r-- | modules/textual_inversion/dataset.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/modules/textual_inversion/dataset.py b/modules/textual_inversion/dataset.py index 67e90afe..12e2f43b 100644 --- a/modules/textual_inversion/dataset.py +++ b/modules/textual_inversion/dataset.py @@ -81,7 +81,8 @@ class PersonalizedBase(Dataset): entry.cond = cond_model([entry.cond_text]).to(devices.cpu)
self.dataset.append(entry)
-
+
+ assert len(self.dataset) > 1, "No images have been found in the dataset."
self.length = len(self.dataset) * repeats
self.initial_indexes = np.arange(self.length) % len(self.dataset)
|