From 24b11ecf9fa01c5da5dc39d62014d8f9e399efab Mon Sep 17 00:00:00 2001
From: Sean McLellan <sean@baristalabs.io>
Date: Tue, 23 Aug 2022 22:45:02 -0400
Subject: [PATCH] Fix silent SystemExit if embedding_path is not specified

---
 ldm/simplet2i.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/ldm/simplet2i.py b/ldm/simplet2i.py
index 25897ed500..6f44d00b99 100644
--- a/ldm/simplet2i.py
+++ b/ldm/simplet2i.py
@@ -436,7 +436,8 @@ The vast majority of these arguments default to reasonable values.
                 config = OmegaConf.load(self.config)
                 self.device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
                 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)
             except AttributeError:
                 raise SystemExit