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

2022-10-14 22:45:26 +02:00 · 2022-10-14 22:45:26 +02:00 · 4d19f3b7d4
commit 4d19f3b7d4
parent 368f4cc4c7
1 changed files with 2 additions and 1 deletions
--- 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)