Fix controlnet issue.

2023-08-31 15:16:58 -04:00 · 2023-08-31 15:16:58 -04:00 · cfe1c54de8
parent 57beace324
commit cfe1c54de8
1 changed files with 1 additions and 1 deletions
--- a/comfy/controlnet.py
+++ b/comfy/controlnet.py
@ -155,7 +155,7 @@ class ControlNet(ControlBase):
            self.cond_hint = broadcast_image_to(self.cond_hint, x_noisy.shape[0], batched_number)


-        context = torch.cat(cond['c_crossattn'], 1)
+        context = cond['c_crossattn']
        y = cond.get('c_adm', None)
        if y is not None:
            y = y.to(self.control_model.dtype)