From cfe1c54de88e7525ec7e4189a8a3294dfc3cd4c0 Mon Sep 17 00:00:00 2001 From: comfyanonymous Date: Thu, 31 Aug 2023 15:16:58 -0400 Subject: [PATCH] Fix controlnet issue. --- comfy/controlnet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/comfy/controlnet.py b/comfy/controlnet.py index 83e1be05..f62dd9c8 100644 --- 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)