Update

1 files changed, 5 insertions, 5 deletions

diff --git a/training/strategy/lora.py b/training/strategy/lora.py
index 3f4dbbc..0c0f633 100644
--- a/training/strategy/lora.py
+++ b/training/strategy/lora.py
@@ -120,11 +120,11 @@ def lora_strategy_callbacks(
         unet_ = accelerator.unwrap_model(unet, keep_fp32_wrapper=False)
         text_encoder_ = accelerator.unwrap_model(text_encoder, keep_fp32_wrapper=False)
 
-        for (token, ids) in zip(placeholder_tokens, placeholder_token_ids):
-            text_encoder_.text_model.embeddings.save_embed(
-                ids,
-                checkpoint_output_dir / f"{slugify(token)}_{step}_{postfix}.bin"
-            )
+        # for (token, ids) in zip(placeholder_tokens, placeholder_token_ids):
+        #     text_encoder_.text_model.embeddings.save_embed(
+        #         ids,
+        #         checkpoint_output_dir / f"{slugify(token)}_{step}_{postfix}.bin"
+        #     )
 
         if not pti_mode:
             lora_config = {}