1 files changed, 1 insertions, 0 deletions

diff --git a/infer.py b/infer.py
index 8fdf63d..4648c0a 100644
--- a/infer.py
+++ b/infer.py
@@ -235,6 +235,7 @@ def load_embeddings(pipeline, embeddings_dir):
         pipeline.text_encoder.text_model.embeddings,
         Path(embeddings_dir)
     )
+    pipeline.text_encoder.text_model.embeddings.persist()
     print(f"Added {len(added_tokens)} tokens from embeddings dir: {list(zip(added_tokens, added_ids))}")