From 46a57e286da77491727114cf335e25fd69f19ac4 Mon Sep 17 00:00:00 2001 From: missionfloyd Date: Wed, 12 Apr 2023 21:25:40 -0600 Subject: [PATCH] Correct height, width variables --- comfy_extras/nodes_post_processing.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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