diff options
| author | Volpeon <git@volpeon.ink> | 2023-01-05 10:26:17 +0100 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2023-01-05 10:26:17 +0100 |
| commit | 4ef37d87d5a04bc6bb7dacee0660bba3057cc02f (patch) | |
| tree | ffcae0352c1b0784b18572f66bba3c617165cef1 /train_dreambooth.py | |
| parent | Various cleanups (diff) | |
| download | textual-inversion-diff-4ef37d87d5a04bc6bb7dacee0660bba3057cc02f.tar.gz textual-inversion-diff-4ef37d87d5a04bc6bb7dacee0660bba3057cc02f.tar.bz2 textual-inversion-diff-4ef37d87d5a04bc6bb7dacee0660bba3057cc02f.zip | |
Fix
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 | ||