From dfedfe40525d6f0564500f4cc27f3905d532eed6 Mon Sep 17 00:00:00 2001 From: Alistair Buxton Date: Mon, 11 Sep 2023 00:34:48 +0100 Subject: [PATCH] Also return batch size from GetImageSize --- image.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/image.py b/image.py index 9f998bc..6458620 100644 --- a/image.py +++ b/image.py @@ -90,13 +90,16 @@ class GetImageSize: }, } - RETURN_TYPES = ("INT", "INT") - RETURN_NAMES = ("width", "height") + RETURN_TYPES = ("INT", "INT", "INT") + RETURN_NAMES = ("width", "height", "batch_size") FUNCTION = "go" CATEGORY = "ali1234/image" def go(self, images): - return (images.shape[2], images.shape[1]) + return (images.shape[2], images.shape[1], images.shape[0]) + + + @register_node