diff --git a/load_image_sequence.py b/load_image_sequence.py index e329d9d..fdf7b92 100644 --- a/load_image_sequence.py +++ b/load_image_sequence.py @@ -31,7 +31,7 @@ class BatchLoadImageSequence: } } - RETURN_TYPES = ("IMAGE",) + RETURN_TYPES = ("IMAGE", "INT") FUNCTION = "execute" def execute( @@ -71,7 +71,7 @@ class BatchLoadImageSequence: imgs.append(img) pbar.update(1) - return (torch.cat(imgs, dim=0),) + return (torch.cat(imgs, dim=0), len(imgs)) NODE_CLASS_MAPPINGS = { diff --git a/save_webp_sequence.py b/save_webp_sequence.py index abc6b2c..e582acd 100644 --- a/save_webp_sequence.py +++ b/save_webp_sequence.py @@ -32,8 +32,8 @@ class BatchSaveImageSequenceWebP: } } - RETURN_TYPES = ("STRING_LIST",) - RETURN_NAMES = ("paths",) + RETURN_TYPES = ("STRING_LIST", "INT") + RETURN_NAMES = ("paths", "count") FUNCTION = "execute" def execute( @@ -147,7 +147,7 @@ class BatchSaveImageSequenceWebP: # Check for exceptions f.result() - return (saved_paths,) + return (saved_paths, len(saved_paths)) NODE_CLASS_MAPPINGS = {