diff options
| author | Volpeon <git@volpeon.ink> | 2023-03-01 12:34:42 +0100 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2023-03-01 12:34:42 +0100 |
| commit | a1b8327085ddeab589be074d7e9df4291aba1210 (patch) | |
| tree | 2f2016916d7a2f659268c3e375d55c59583c2b3b /models | |
| parent | Fixed TI normalization order (diff) | |
| download | textual-inversion-diff-a1b8327085ddeab589be074d7e9df4291aba1210.tar.gz textual-inversion-diff-a1b8327085ddeab589be074d7e9df4291aba1210.tar.bz2 textual-inversion-diff-a1b8327085ddeab589be074d7e9df4291aba1210.zip | |
Update
Diffstat (limited to 'models')
| -rw-r--r-- | models/clip/embeddings.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/models/clip/embeddings.py b/models/clip/embeddings.py index 734730e..6be6e9f 100644 --- a/models/clip/embeddings.py +++ b/models/clip/embeddings.py | |||
| @@ -13,7 +13,7 @@ from transformers.models.clip.modeling_clip import CLIPTextEmbeddings | |||
| 13 | 13 | ||
| 14 | 14 | ||
| 15 | def resize_embedding(old_embedding: nn.Embedding, new_num_embeddings: int, initializer_factor: float = 1.0) -> nn.Embedding: | 15 | def resize_embedding(old_embedding: nn.Embedding, new_num_embeddings: int, initializer_factor: float = 1.0) -> nn.Embedding: |
| 16 | old_num_embeddings, old_embedding_dim = old_embedding.weight.size() | 16 | old_num_embeddings, old_embedding_dim = old_embedding.weight.shape |
| 17 | 17 | ||
| 18 | if old_num_embeddings == new_num_embeddings: | 18 | if old_num_embeddings == new_num_embeddings: |
| 19 | return old_embedding | 19 | return old_embedding |