فهرست منبع

Better error handling when unable to extract contents of embedding/TI file

Brendan Hoar 1 سال پیش
والد
کامیت
c5b7559856
1فایلهای تغییر یافته به همراه8 افزوده شده و 4 حذف شده
  1. 8 4
      modules/textual_inversion/textual_inversion.py

+ 8 - 4
modules/textual_inversion/textual_inversion.py

@@ -181,12 +181,16 @@ class EmbeddingDatabase:
         else:
             return
 
-        embedding = create_embedding_from_data(data, name, filename=filename, filepath=path)
+        if data is not None:
+            embedding = create_embedding_from_data(data, name, filename=filename, filepath=path)
 
-        if self.expected_shape == -1 or self.expected_shape == embedding.shape:
-            self.register_embedding(embedding, shared.sd_model)
+            if self.expected_shape == -1 or self.expected_shape == embedding.shape:
+                self.register_embedding(embedding, shared.sd_model)
+            else:
+                self.skipped_embeddings[name] = embedding
         else:
-            self.skipped_embeddings[name] = embedding
+            print(f"Unable to load Textual inversion embedding due to data issue: '{name}'.")
+
 
     def load_from_dir(self, embdir):
         if not os.path.isdir(embdir.path):