1 files changed, 3 insertions, 3 deletions

diff --git a/train_ti.py b/train_ti.py
index 89f4113..1d0cb6f 100644
--- a/train_ti.py
+++ b/train_ti.py
@@ -691,7 +691,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))}"
         )
@@ -712,7 +712,7 @@ def main():
             args.placeholder_tokens = added_tokens
             print("Training embeddings from embeddings dir")
         else:
-            embeddings.persist()
+            embeddings.persist(True)
 
     if args.scale_lr:
         args.learning_rate = (
@@ -1067,7 +1067,7 @@ def main():
             args.train_data_template,
         ):
             run(i, [placeholder_token], [initializer_token], num_vectors, data_template)
-            embeddings.persist()
+            embeddings.persist(True)
 
 
 if __name__ == "__main__":