diff --git a/comfy/ldm/modules/diffusionmodules/openaimodel.py b/comfy/ldm/modules/diffusionmodules/openaimodel.py index 4c8d53cac..2173efd47 100644 --- a/comfy/ldm/modules/diffusionmodules/openaimodel.py +++ b/comfy/ldm/modules/diffusionmodules/openaimodel.py @@ -367,6 +367,8 @@ def apply_control(h, control, name): ctrl = control[name].pop() if ctrl is not None: try: + if ctrl.shape[2] != h.shape[2] or ctrl.shape[3] != h.shape[3]: + ctrl = F.interpolate(ctrl.float(), size=(h.shape[2], h.shape[3]), mode="bicubic", align_corners=False).to(h.dtype) h += ctrl except: logging.warning("warning control could not be applied {} {}".format(h.shape, ctrl.shape))