Commit c5b75598 authored by Brendan Hoar's avatar Brendan Hoar Committed by GitHub

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

parent 8dc92022
...@@ -181,12 +181,16 @@ class EmbeddingDatabase: ...@@ -181,12 +181,16 @@ class EmbeddingDatabase:
else: else:
return 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: if self.expected_shape == -1 or self.expected_shape == embedding.shape:
self.register_embedding(embedding, shared.sd_model) self.register_embedding(embedding, shared.sd_model)
else:
self.skipped_embeddings[name] = embedding
else: 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): def load_from_dir(self, embdir):
if not os.path.isdir(embdir.path): if not os.path.isdir(embdir.path):
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment