Add INT count output to all nodes
This commit is contained in:
@@ -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 = {
|
||||
|
||||
@@ -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 = {
|
||||
|
||||
Reference in New Issue
Block a user