xinyun/ComfyUI-KJNodes
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-KJNodes.git synced 2026-05-27 04:45:42 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update mask_nodes.py

Browse Source
This commit is contained in:
kijai 2025-10-03 19:35:15 +03:00
parent fb253119ed
commit 625151d43f
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
nodes/mask_nodes.py
View File

@ -1603,7 +1603,6 @@ class BlockifyMask:
def process(self, masks, block_size, device="cpu"): def process(self, masks, block_size, device="cpu"):
processing_device = main_device if device == "gpu" else torch.device("cpu") processing_device = main_device if device == "gpu" else torch.device("cpu")
print("mask.dtype:", masks.dtype, "device:", processing_device)
masks = masks.to(processing_device) masks = masks.to(processing_device)
batch_size, height, width = masks.shape batch_size, height, width = masks.shape