Skip to content

Commit a27d19d

Browse files
committed
Additional assert on dataset
1 parent 3ce2bfd commit a27d19d

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

modules/textual_inversion/dataset.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -42,6 +42,8 @@ def __init__(self, data_root, width, height, repeats, flip_p=0.5, placeholder_to
4242
self.lines = lines
4343

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

4648
cond_model = shared.sd_model.cond_stage_model
4749

0 commit comments

Comments
 (0)