From 12b9aca96a36dd77a6b2b99bbc1743d87a7ce733 Mon Sep 17 00:00:00 2001
From: Volpeon <git@volpeon.ink>
Date: Sat, 24 Jun 2023 21:00:29 +0200
Subject: Update

---
 train_lora.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

(limited to 'train_lora.py')

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
     # --------------------------------------------------------------------------------
-- 
cgit v1.2.3-70-g09d2