diff --git a/comfy_extras/nodes_post_processing.py b/comfy_extras/nodes_post_processing.py index 1d554422b..293df28d7 100644 --- a/comfy_extras/nodes_post_processing.py +++ b/comfy_extras/nodes_post_processing.py @@ -320,8 +320,8 @@ class Rotate: color = fill_color.replace(" ", "") color = parse_palette(color) rotated_image = pil_image.rotate(angle=angle, resample=resamplers[resample], expand=expand, center=center, translate=translate, fillcolor=color) - height, width = rotated_image.size - result = torch.zeros(batch_size, width, height, 3) + width, height = rotated_image.size + result = torch.zeros(batch_size, height, width, 3) rotated_array = torch.tensor(np.array(rotated_image.convert("RGB"))).float() / 255 result[0] = rotated_array