Update

1 files changed, 3 insertions, 3 deletions

diff --git a/train_ti.py b/train_ti.py
index 1b8c597..0ffc9e6 100644
--- a/train_ti.py
+++ b/train_ti.py
@@ -607,10 +607,10 @@ def main():
     new_ids = tokenizer.add_multi_tokens(args.placeholder_token, args.num_vectors)
     embeddings.resize(len(tokenizer))
 
-    init_ratios = [
+    for (new_id, init_ids) in zip(new_ids, initializer_token_ids):
         embeddings.add_embed(new_id, init_ids)
-        for (new_id, init_ids) in zip(new_ids, initializer_token_ids)
-    ]
+
+    init_ratios = [f"{len(init_ids)} / {len(new_id)}" for new_id, init_ids in zip(new_ids, initializer_token_ids)]
 
     print(f"Added {len(new_ids)} new tokens: {list(zip(args.placeholder_token, new_ids, init_ratios))}")