diff options
| author | Volpeon <git@volpeon.ink> | 2022-12-31 14:07:44 +0100 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2022-12-31 14:07:44 +0100 |
| commit | dc463a6b8ef120b7a0643569b66f9109ed38c652 (patch) | |
| tree | ae742a988b8541009a980c8b2f719696f9d7df27 /common.py | |
| parent | Fixes (diff) | |
| download | textual-inversion-diff-dc463a6b8ef120b7a0643569b66f9109ed38c652.tar.gz textual-inversion-diff-dc463a6b8ef120b7a0643569b66f9109ed38c652.tar.bz2 textual-inversion-diff-dc463a6b8ef120b7a0643569b66f9109ed38c652.zip | |
Simplified multi-vector embedding code
Diffstat (limited to 'common.py')
| -rw-r--r-- | common.py | 3 |
1 files changed, 1 insertions, 2 deletions
| @@ -32,7 +32,6 @@ def load_embeddings_from_dir(tokenizer: MultiCLIPTokenizer, embeddings: ManagedC | |||
| 32 | embed = file.get_tensor("embed") | 32 | embed = file.get_tensor("embed") |
| 33 | 33 | ||
| 34 | added = tokenizer.add_multi_tokens(filename.stem, embed.shape[0]) | 34 | added = tokenizer.add_multi_tokens(filename.stem, embed.shape[0]) |
| 35 | embeddings.add_embed(added.placeholder_id) | 35 | embeddings.add_embed(added.ids, embed) |
| 36 | embeddings.add_embed(added.multi_ids, embed) | ||
| 37 | 36 | ||
| 38 | return tokens | 37 | return tokens |