1 files changed, 1 insertions, 0 deletions

diff --git a/train_dreambooth.py b/train_dreambooth.py
index f4d4cbb..2aca1e7 100644
--- a/train_dreambooth.py
+++ b/train_dreambooth.py
@@ -513,6 +513,7 @@ def main():
             raise ValueError("--embeddings_dir must point to an existing directory")
 
         added_tokens, added_ids = load_embeddings_from_dir(tokenizer, embeddings, embeddings_dir)
+        embeddings.persist()
         print(f"Added {len(added_tokens)} tokens from embeddings dir: {list(zip(added_tokens, added_ids))}")
 
     if args.scale_lr: