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():
             placeholder_tokens=alias_placeholder_tokens,
             initializer_tokens=alias_initializer_tokens,
         )
-        embeddings.persist()
+        embeddings.persist(True)
         print(
             f"Added {len(added_tokens)} aliases: {list(zip(alias_placeholder_tokens, added_tokens, alias_initializer_tokens, added_ids))}"
         )
@@ -682,7 +682,7 @@ def main():
             f"Added {len(added_tokens)} tokens from embeddings dir: {list(zip(added_tokens, added_ids))}"
         )
 
-        embeddings.persist()
+        embeddings.persist(True)
 
     if len(args.placeholder_tokens) != 0:
         placeholder_token_ids, initializer_token_ids = add_placeholder_tokens(