xinyun/ComfyUI-KJNodes
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-KJNodes.git synced 2026-07-19 04:07:10 +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:14:27 +03:00
parent d1f6cf77b0
commit 4d411c4593
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/mask_nodes.py
View File

@ -1013,7 +1013,7 @@ class GrowMaskWithBlur:
out = [] out = []
previous_output = None previous_output = None
current_expand = expand current_expand = expand
for m in growmask: for m in tqdm(growmask, desc="Expanding/Contracting Mask"):
output = m.unsqueeze(0).unsqueeze(0).to(main_device) # Add batch and channel dims for kornia output = m.unsqueeze(0).unsqueeze(0).to(main_device) # Add batch and channel dims for kornia
if abs(round(current_expand)) > 0: if abs(round(current_expand)) > 0:
# Create kernel - kornia expects kernel on same device as input # Create kernel - kornia expects kernel on same device as input