Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 01:04:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

register skipped embeddings correctly

Browse Source
This commit is contained in:
Vladimir Mandic
2023-09-26 07:04:14 -04:00
parent c8b106a471
commit db6091207d
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/textual_inversion/textual_inversion.py
+1 -1
View File
@@ -193,7 +193,7 @@ class EmbeddingDatabase:
# diffuser concepts
elif type(data) == dict and type(next(iter(data.values()))) == torch.Tensor:
if len(data.keys()) != 1:
self.skipped_embeddings[name] = Embedding(None, name)
self.skipped_embeddings[name] = Embedding(None, name=name, filename=path)
return
emb = next(iter(data.values()))
if len(emb.shape) == 1: