diff options
Diffstat (limited to 'train_dreambooth.py')
-rw-r--r-- | train_dreambooth.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/train_dreambooth.py b/train_dreambooth.py index c658ad6..1fd86b1 100644 --- a/train_dreambooth.py +++ b/train_dreambooth.py | |||
@@ -612,7 +612,7 @@ def main(): | |||
612 | raise ValueError("--embeddings_dir must point to an existing directory") | 612 | raise ValueError("--embeddings_dir must point to an existing directory") |
613 | 613 | ||
614 | added_tokens, added_ids = load_embeddings_from_dir(tokenizer, embeddings, embeddings_dir) | 614 | added_tokens, added_ids = load_embeddings_from_dir(tokenizer, embeddings, embeddings_dir) |
615 | print(f"Added {len(added_tokens)} tokens from embeddings dir: {zip(added_tokens, added_ids)}") | 615 | print(f"Added {len(added_tokens)} tokens from embeddings dir: {list(zip(added_tokens, added_ids))}") |
616 | 616 | ||
617 | if len(args.placeholder_token) != 0: | 617 | if len(args.placeholder_token) != 0: |
618 | # Convert the initializer_token, placeholder_token to ids | 618 | # Convert the initializer_token, placeholder_token to ids |
@@ -629,7 +629,7 @@ def main(): | |||
629 | for (new_id, init_ids) in zip(new_ids, initializer_token_ids) | 629 | for (new_id, init_ids) in zip(new_ids, initializer_token_ids) |
630 | ] | 630 | ] |
631 | 631 | ||
632 | print(f"Added {len(new_ids)} new tokens: {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: |
634 | placeholder_token_id = [] | 634 | placeholder_token_id = [] |
635 | 635 | ||