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 beb65fc..929310b 100644 --- a/train_dreambooth.py +++ b/train_dreambooth.py | |||
@@ -661,7 +661,7 @@ def main(): | |||
661 | placeholder_tokens=alias_placeholder_tokens, | 661 | placeholder_tokens=alias_placeholder_tokens, |
662 | initializer_tokens=alias_initializer_tokens, | 662 | initializer_tokens=alias_initializer_tokens, |
663 | ) | 663 | ) |
664 | embeddings.persist() | 664 | embeddings.persist(True) |
665 | print( | 665 | print( |
666 | f"Added {len(added_tokens)} aliases: {list(zip(alias_placeholder_tokens, added_tokens, alias_initializer_tokens, added_ids))}" | 666 | f"Added {len(added_tokens)} aliases: {list(zip(alias_placeholder_tokens, added_tokens, alias_initializer_tokens, added_ids))}" |
667 | ) | 667 | ) |
@@ -682,7 +682,7 @@ def main(): | |||
682 | f"Added {len(added_tokens)} tokens from embeddings dir: {list(zip(added_tokens, added_ids))}" | 682 | f"Added {len(added_tokens)} tokens from embeddings dir: {list(zip(added_tokens, added_ids))}" |
683 | ) | 683 | ) |
684 | 684 | ||
685 | embeddings.persist() | 685 | embeddings.persist(True) |
686 | 686 | ||
687 | if len(args.placeholder_tokens) != 0: | 687 | if len(args.placeholder_tokens) != 0: |
688 | placeholder_token_ids, initializer_token_ids = add_placeholder_tokens( | 688 | placeholder_token_ids, initializer_token_ids = add_placeholder_tokens( |