1 files changed, 1 insertions, 2 deletions

diff --git a/train_ti.py b/train_ti.py
index 69d15ea..3a5cfed 100644
--- a/train_ti.py
+++ b/train_ti.py
@@ -439,11 +439,10 @@ class Checkpointer(CheckpointerBase):
         for new_token in self.new_tokens:
             text_encoder.text_model.embeddings.save_embed(
                 new_token.multi_ids,
-                f"{slugify(new_token.token)}_{step}_{postfix}.bin"
+                checkpoints_path.joinpath(f"{slugify(new_token.token)}_{step}_{postfix}.bin")
             )
 
         del text_encoder
-        del learned_embeds
 
     @torch.no_grad()
     def save_samples(self, step, num_inference_steps, guidance_scale=7.5, eta=0.0):