diff options
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 |