diff --git a/nodes/image_nodes.py b/nodes/image_nodes.py index 7cb9cae..3660d43 100644 --- a/nodes/image_nodes.py +++ b/nodes/image_nodes.py @@ -1137,7 +1137,13 @@ class ImagePadForOutpaintTargetSize: else: # If downscaling is not needed, use the original image dimensions image_scaled = image - mask_scaled = mask + if mask is not None: + # Ensure mask dimensions match image dimensions + mask_scaled = mask.unsqueeze(0) # Add an extra dimension for batch size + mask_scaled = F.interpolate(mask_scaled, size=(new_height, new_width), mode="nearest") + mask_scaled = mask_scaled.squeeze(0) # Remove the extra dimension after interpolation + else: + mask_scaled = mask # Calculate how much padding is needed to reach the target dimensions pad_top = max(0, (target_height - new_height) // 2)