Fix multiple controlnets not working.

comfyanonymous · Feb 26, 2023 · dfb397e · dfb397e
1 parent 6de6246
commit dfb397e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/comfy/sd.py b/comfy/sd.py
@@ -393,7 +393,7 @@ def __init__(self, control_model, device="cuda"):
     def get_control(self, x_noisy, t, cond_txt, batched_number):
         control_prev = None
         if self.previous_controlnet is not None:
-            control_prev = self.previous_controlnet.get_control(x_noisy, t, cond_txt)
+            control_prev = self.previous_controlnet.get_control(x_noisy, t, cond_txt, batched_number)
 
         output_dtype = x_noisy.dtype
         if self.cond_hint is None or x_noisy.shape[2] * 8 != self.cond_hint.shape[2] or x_noisy.shape[3] * 8 != self.cond_hint.shape[3]:
@@ -541,7 +541,7 @@ def __init__(self, t2i_model, channels_in, device="cuda"):
     def get_control(self, x_noisy, t, cond_txt, batched_number):
         control_prev = None
         if self.previous_controlnet is not None:
-            control_prev = self.previous_controlnet.get_control(x_noisy, t, cond_txt)
+            control_prev = self.previous_controlnet.get_control(x_noisy, t, cond_txt, batched_number)
 
         if self.cond_hint is None or x_noisy.shape[2] * 8 != self.cond_hint.shape[2] or x_noisy.shape[3] * 8 != self.cond_hint.shape[3]:
             if self.cond_hint is not None: