From 7b04d813739c0b5595295dffdc86cc41108db2d3 Mon Sep 17 00:00:00 2001 From: Volpeon Date: Sat, 6 May 2023 16:25:36 +0200 Subject: Update --- training/strategy/lora.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) (limited to 'training/strategy/lora.py') 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 = {} -- cgit v1.2.3-54-g00ecf