We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 7337b87 commit ee8318eCopy full SHA for ee8318e
loaders.py
@@ -555,7 +555,7 @@ def load_textual_inversion(
555
embeddings = [e for e in embedding] # noqa: C416
556
else:
557
tokens = [token]
558
- embeddings = [embedding] if len(embedding.shape) > 1 else [embedding[0]]
+ embeddings = [embedding[0]] if len(embedding.shape) > 1 else [embedding]
559
560
# add tokens and get ids
561
self.tokenizer.add_tokens(tokens)
0 commit comments