mirror of
https://git.datalinker.icu/kijai/ComfyUI-KJNodes.git
synced 2025-12-09 04:44:30 +08:00
Update mask_nodes.py
This commit is contained in:
parent
fb253119ed
commit
625151d43f
@ -1603,7 +1603,6 @@ class BlockifyMask:
|
||||
|
||||
def process(self, masks, block_size, 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)
|
||||
batch_size, height, width = masks.shape
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user