From 73f7ef521467b2b63a63afc7d172507699e796eb Mon Sep 17 00:00:00 2001 From: bymyself Date: Fri, 16 May 2025 14:04:54 -0700 Subject: [PATCH] fix rebase error --- comfy_api_nodes/util/validation_utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/comfy_api_nodes/util/validation_utils.py b/comfy_api_nodes/util/validation_utils.py index 2b5bbdf86..031b9fbd3 100644 --- a/comfy_api_nodes/util/validation_utils.py +++ b/comfy_api_nodes/util/validation_utils.py @@ -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.")