Update

1 files changed, 3 insertions, 3 deletions

diff --git a/train_dreambooth.py b/train_dreambooth.py
index c355ea1..e8256be 100644
--- a/train_dreambooth.py
+++ b/train_dreambooth.py
@@ -624,10 +624,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))}")
     else: