diff --git a/comfy/ldm/flux/model.py b/comfy/ldm/flux/model.py index c4de82795b17..f84475bd905d 100644 --- a/comfy/ldm/flux/model.py +++ b/comfy/ldm/flux/model.py @@ -157,6 +157,11 @@ def block_wrap(args): if i < len(control_i): add = control_i[i] if add is not None: + if img.shape[1] != add.shape[1]: + padding_size = img.shape[1] - add.shape[1] + if padding_size > 0: + padding = torch.zeros(add.shape[0], padding_size, add.shape[2], device=add.device, dtype=add.dtype) + add = torch.cat([add, padding], dim=1) img += add if img.dtype == torch.float16: