Print warning when controlnet can't be applied instead of crashing.

comfyanonymous · Nov 16, 2023 · 7ea6bb0 · 7ea6bb0
1 parent dcec104
commit 7ea6bb0
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/comfy/ldm/modules/diffusionmodules/openaimodel.py b/comfy/ldm/modules/diffusionmodules/openaimodel.py
@@ -255,7 +255,10 @@ def apply_control(h, control, name):
     if control is not None and name in control and len(control[name]) > 0:
         ctrl = control[name].pop()
         if ctrl is not None:
-            h += ctrl
+            try:
+                h += ctrl
+            except:
+                print("warning control could not be applied", h.shape, ctrl.shape)
     return h
 
 class UNetModel(nn.Module):