summaryrefslogtreecommitdiffstats
path: root/common.py
diff options
context:
space:
mode:
Diffstat (limited to 'common.py')
-rw-r--r--common.py3
1 files changed, 1 insertions, 2 deletions
diff --git a/common.py b/common.py
index e8d3ac1..1e7f4b9 100644
--- a/common.py
+++ b/common.py
@@ -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