wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-05-13 10:42:59 +08:00
Code Issues Actions 15 Packages Projects Releases Wiki Activity

Correct height, width variables

Browse Source
This commit is contained in:
missionfloyd 2023-04-12 21:25:40 -06:00 committed by GitHub
parent f5474109a4
commit 46a57e286d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
comfy_extras/nodes_post_processing.py
View File

@ -320,8 +320,8 @@ class Rotate:
color = fill_color.replace(" ", "") color = fill_color.replace(" ", "")
color = parse_palette(color) color = parse_palette(color)
rotated_image = pil_image.rotate(angle=angle, resample=resamplers[resample], expand=expand, center=center, translate=translate, fillcolor=color) rotated_image = pil_image.rotate(angle=angle, resample=resamplers[resample], expand=expand, center=center, translate=translate, fillcolor=color)
height, width = rotated_image.size width, height = rotated_image.size
result = torch.zeros(batch_size, width, height, 3) result = torch.zeros(batch_size, height, width, 3)
rotated_array = torch.tensor(np.array(rotated_image.convert("RGB"))).float() / 255 rotated_array = torch.tensor(np.array(rotated_image.convert("RGB"))).float() / 255
result[0] = rotated_array result[0] = rotated_array