diff --git a/nodes/mask_nodes.py b/nodes/mask_nodes.py index 72d7d45..7459302 100644 --- a/nodes/mask_nodes.py +++ b/nodes/mask_nodes.py @@ -1013,7 +1013,7 @@ class GrowMaskWithBlur: out = [] previous_output = None 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 if abs(round(current_expand)) > 0: # Create kernel - kornia expects kernel on same device as input