Additional assert on dataset

a27d19de · Muhammad Rizqi Nur · 3ce2bfdf · a27d19de
Commit a27d19de authored Oct 29, 2022 by Muhammad Rizqi Nur
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

modules/textual_inversion/dataset.py modules/textual_inversion/dataset.py +2 -0

No files found.
--- a/modules/textual_inversion/dataset.py
+++ b/modules/textual_inversion/dataset.py
@@ -42,6 +42,8 @@ class PersonalizedBase(Dataset):
        self.lines = lines

        assert data_root, 'dataset directory not specified'
+        assert os.path.isdir(data_root), "Dataset directory doesn't exist"
+        assert os.listdir(data_root), "Dataset directory is empty"

        cond_model = shared.sd_model.cond_stage_model