Update mask_nodes.py

This commit is contained in:
kijai 2025-10-03 19:35:15 +03:00
parent fb253119ed
commit 625151d43f

View File

@ -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