From 87d0cf42db7d59992daba4d58a83655b5b359f44 Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Wed, 6 Aug 2025 23:15:12 +0300
Subject: [PATCH] fix dit lora block loader on loras with diff keys

---
 nodes/nodes.py | 25 ++++++++++++++++---------
 1 file changed, 16 insertions(+), 9 deletions(-)

diff --git a/nodes/nodes.py b/nodes/nodes.py
index 73cc839..dbbd185 100644
--- a/nodes/nodes.py
+++ b/nodes/nodes.py
@@ -1935,22 +1935,29 @@ class DiTBlockLoraLoader:
                         if ratio == 0:
                             keys_to_delete.append(key)
                         else:
-                            value = loaded[key].weights
-                            weights_list = list(loaded[key].weights)
-                            weights_list[2] = ratio
-                            loaded[key].weights = tuple(weights_list)
+                            # Only modify LoRA adapters, skip diff tuples
+                            value = loaded[key]
+                            if hasattr(value, 'weights'):
+                                print(f"Modifying LoRA adapter for key: {key}")
+                                weights_list = list(value.weights)
+                                weights_list[2] = ratio
+                                loaded[key].weights = tuple(weights_list)
+                            else:
+                                print(f"Skipping non-LoRA entry for key: {key}")
 
             for key in keys_to_delete:
                 del loaded[key]
 
             print("loading lora keys:")
             for key, value in loaded.items():
-                print(f"Key: {key}, Alpha: {value.weights[2]}")
+                if hasattr(value, 'weights'):
+                    print(f"Key: {key}, Alpha: {value.weights[2]}")
+                else:
+                    print(f"Key: {key}, Type: {type(value)}")
 
-
-                if model is not None:
-                    new_modelpatcher = model.clone()
-                    k = new_modelpatcher.add_patches(loaded, strength_model)  
+        if model is not None:
+            new_modelpatcher = model.clone()
+            k = new_modelpatcher.add_patches(loaded, strength_model)  
     
         k = set(k)
         for x in loaded: