diff options
Diffstat (limited to 'train_dreambooth.py')
-rw-r--r-- | train_dreambooth.py | 6 |
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(): | |||
624 | new_ids = tokenizer.add_multi_tokens(args.placeholder_token, args.num_vectors) | 624 | new_ids = tokenizer.add_multi_tokens(args.placeholder_token, args.num_vectors) |
625 | embeddings.resize(len(tokenizer)) | 625 | embeddings.resize(len(tokenizer)) |
626 | 626 | ||
627 | init_ratios = [ | 627 | for (new_id, init_ids) in zip(new_ids, initializer_token_ids): |
628 | embeddings.add_embed(new_id, init_ids) | 628 | embeddings.add_embed(new_id, init_ids) |
629 | for (new_id, init_ids) in zip(new_ids, initializer_token_ids) | 629 | |
630 | ] | 630 | init_ratios = [f"{len(init_ids)} / {len(new_id)}" for new_id, init_ids in zip(new_ids, initializer_token_ids)] |
631 | 631 | ||
632 | print(f"Added {len(new_ids)} new tokens: {list(zip(args.placeholder_token, new_ids, init_ratios))}") | 632 | print(f"Added {len(new_ids)} new tokens: {list(zip(args.placeholder_token, new_ids, init_ratios))}") |
633 | else: | 633 | else: |