From 345c286c867440978a0ce4c2311724844606c8f4 Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Tue, 19 Aug 2025 16:55:51 +0300
Subject: [PATCH] Update utils.py

---
 utils.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/utils.py b/utils.py
index d824fec..85d8637 100644
--- a/utils.py
+++ b/utils.py
@@ -210,7 +210,7 @@ def apply_lora(model, device_to, transformer_load_device, params_to_keep=None, d
             pbar.update(1)
         
         # After LoRA patching, scale weights that have scale_weight but are NOT LoRA patched
-        if len(scale_weights) > 0:
+        if len(scale_weights) > 0 and not getattr(model, "scale_weights_applied", False):
             for name, param in model.model.diffusion_model.named_parameters():
                 scale_key = name.replace("weight", "scale_weight").replace("diffusion_model.", "") if "weight" in name else None
                 full_param_name = f"diffusion_model.{name}"
@@ -219,6 +219,7 @@ def apply_lora(model, device_to, transformer_load_device, params_to_keep=None, d
                     param_fp32 = param.to(torch.float32)
                     param_fp32.mul_(scale.to(param.device, torch.float32))
                     param.copy_(param_fp32.to(param.dtype))
+            model.scale_weights_applied = True
       
         model.current_weight_patches_uuid = model.patches_uuid
         if low_mem_load: