Skip to content

Commit 1055175

Browse files
authored
Fix textual inversion loading (#2914)
1 parent 0df4ad5 commit 1055175

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/diffusers/loaders.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -555,7 +555,7 @@ def load_textual_inversion(
555555
embeddings = [e for e in embedding] # noqa: C416
556556
else:
557557
tokens = [token]
558-
embeddings = [embedding] if len(embedding.shape) > 1 else [embedding[0]]
558+
embeddings = [embedding[0]] if len(embedding.shape) > 1 else [embedding]
559559

560560
# add tokens and get ids
561561
self.tokenizer.add_tokens(tokens)

0 commit comments

Comments
 (0)