From e241998f53ad1495169a8fede76b508d583a2def Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Wed, 27 Aug 2025 12:21:25 +0300
Subject: [PATCH] Fix Uni3C

---
 nodes.py               | 2 ++
 nodes_model_loading.py | 2 +-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/nodes.py b/nodes.py
index b25b7c9..20812ca 100644
--- a/nodes.py
+++ b/nodes.py
@@ -45,6 +45,8 @@ def offload_transformer(transformer):
     
     if transformer.patched_linear:
         for name, param in transformer.named_parameters():
+            if "controlnet" in name:
+                continue
             module = transformer
             subnames = name.split('.')
             for subname in subnames[:-1]:
diff --git a/nodes_model_loading.py b/nodes_model_loading.py
index a5f6567..57803b0 100644
--- a/nodes_model_loading.py
+++ b/nodes_model_loading.py
@@ -804,7 +804,7 @@ def load_weights(transformer, sd=None, weight_dtype=None, base_dtype=None,
             except Exception:
                 block_idx = None
 
-        if "loras" in name:
+        if "loras" in name or "controlnet" in name:
             continue
 
         # GGUF: skip GGUFParameter params