xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-09-04 10:37:08 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix rebase error

Browse Source
This commit is contained in:
bymyself 2025-05-16 14:04:54 -07:00
parent 9b62ebd817
commit 73f7ef5214
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_api_nodes/util/validation_utils.py
View File

@ -7,9 +7,9 @@ from comfy_api.input.video_types import VideoInput
def get_image_dimensions(image: torch.Tensor) -> tuple[int, int]:
if len(image.shape) == 4:
height, width = image.shape[1], image.shape[2]
return image.shape[1], image.shape[2]
elif len(image.shape) == 3:
height, width = image.shape[0], image.shape[1]
return image.shape[0], image.shape[1]
else:
raise ValueError("Invalid image tensor shape.")