Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix silent SystemExit if embedding_path is not specified

Browse Source
This commit is contained in:
Sean McLellan
2022-08-23 22:45:02 -04:00
parent 84989f0d05
commit 24b11ecf9f
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
ldm/simplet2i.py
View File

@ -436,7 +436,8 @@ The vast majority of these arguments default to reasonable values.
config = OmegaConf.load(self.config) config = OmegaConf.load(self.config)
self.device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu") self.device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
model = self._load_model_from_config(config,self.weights) model = self._load_model_from_config(config,self.weights)
model.embedding_manager.load(self.embedding_path) if not self.embedding_path is None:
model.embedding_manager.load(self.embedding_path)
self.model = model.to(self.device) self.model = model.to(self.device)
except AttributeError: except AttributeError:
raise SystemExit raise SystemExit