Update

1 files changed, 3 insertions, 3 deletions

diff --git a/train_lora.py b/train_lora.py
index 2a43252..eeac81f 100644
--- a/train_lora.py
+++ b/train_lora.py
@@ -777,7 +777,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))}"
         )
@@ -806,7 +806,7 @@ def main():
         if args.train_dir_embeddings:
             print("Training embeddings from embeddings dir")
         else:
-            embeddings.persist()
+            embeddings.persist(True)
 
     if len(args.placeholder_tokens) != 0 and not args.train_dir_embeddings:
         embeddings = ensure_embeddings()
@@ -1117,7 +1117,7 @@ def main():
             no_val=True,
         )
 
-        embeddings.persist()
+        embeddings.persist(True)
 
     # LORA
     # --------------------------------------------------------------------------------