Additional assert on dataset (a27d19de) · Commits · github_fork / Stable Diffusion Webui

modules/textual_inversion/dataset.py

+2 −0

Original line number	Original line	Diff line number	Diff line
	@@ -42,6 +42,8 @@ class PersonalizedBase(Dataset):
	self.lines = lines		self.lines = lines

	assert data_root, 'dataset directory not specified'		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		cond_model = shared.sd_model.cond_stage_model