AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-29 07:22:12 +03:00
Code Activity

Fix textual inversion loading (#2914)

Browse Source
This commit is contained in:
Guillermo Cique
2023-03-31 10:52:48 +02:00
committed by GitHub
parent 0df4ad541f
commit 1055175a18
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/diffusers/loaders.py
View File

@@ -555,7 +555,7 @@ class TextualInversionLoaderMixin:
embeddings = [e for e in embedding] # noqa: C416
else:
tokens = [token]
embeddings = [embedding] if len(embedding.shape) > 1 else [embedding[0]]
embeddings = [embedding[0]] if len(embedding.shape) > 1 else [embedding]
# add tokens and get ids
self.tokenizer.add_tokens(tokens)