xinyun/ComfyUI-KJNodes
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-KJNodes.git synced 2026-04-13 06:46:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

Node FilterZeroMasksAndCorrespondingImages Edge case handle

Browse Source
This commit is contained in:
MrForExample 2024-01-02 20:27:46 +01:00
parent 2830476fd4
commit ccd89290ed
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
nodes.py
View File

@ -1864,7 +1864,7 @@ class FilterZeroMasksAndCorrespondingImages:
if also_process_images:
non_zero_mask_images_out = torch.stack(non_zero_mask_images, dim=0)
zero_mask_images_out = torch.stack(zero_mask_images, dim=0)
zero_mask_images_out = torch.stack(zero_mask_images, dim=0) if len(zero_mask_images) > 0 else None
else:
non_zero_mask_images_out = zero_mask_images_out = None