Fix Uni3C

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]:

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