Additional assert on dataset

2022-10-29 19:44:05 +07:00 · 2022-10-29 19:44:05 +07:00 · a27d19de2e
commit a27d19de2e
parent 3ce2bfdf95
1 changed files with 2 additions and 0 deletions
--- 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