Fix issue when using multiple t2i adapters with batched images.

10 months ago · 1a57423d30
1 changed files with 4 additions and 1 deletions
--- a/comfy/controlnet.py
+++ b/comfy/controlnet.py
@ -126,7 +126,10 @@ class ControlBase:
                        if o[i] is None:
                            o[i] = prev_val
                        else:
-                            o[i] += prev_val
+                            if o[i].shape[0] < prev_val.shape[0]:
+                                o[i] = prev_val + o[i]
+                            else:
+                                o[i] += prev_val
        return out

 class ControlNet(ControlBase):