From c025308918b57c9a57c67cf066495169b3814f1a Mon Sep 17 00:00:00 2001
From: Kohaku-Blueleaf <59680068+KohakuBlueleaf@users.noreply.github.com>
Date: Tue, 22 Jul 2025 16:34:11 +0800
Subject: [PATCH] correct key name for loha

---
 comfy/weight_adapter/loha.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/comfy/weight_adapter/loha.py b/comfy/weight_adapter/loha.py
index 1a73768a7..cf11e7586 100644
--- a/comfy/weight_adapter/loha.py
+++ b/comfy/weight_adapter/loha.py
@@ -13,10 +13,10 @@ class LohaDiff(WeightAdapterTrainBase):
         w1a, w1b, alpha, w2a, w2b, t1, t2, dora_scale = weights
 
         # Create trainable parameters
-        self.w1a = torch.nn.Parameter(w1a)
-        self.w1b = torch.nn.Parameter(w1b)
-        self.w2a = torch.nn.Parameter(w2a)
-        self.w2b = torch.nn.Parameter(w2b)
+        self.w1_a = torch.nn.Parameter(w1a)
+        self.w1_b = torch.nn.Parameter(w1b)
+        self.w2_a = torch.nn.Parameter(w2a)
+        self.w2_b = torch.nn.Parameter(w2b)
 
         self.use_tucker = False
         if t1 is not None and t2 is not None: