From 18fba2a3c0ec9fa4920737ebd16e8e123171a67f Mon Sep 17 00:00:00 2001 From: thot-experiment Date: Wed, 23 Apr 2025 11:50:26 -0700 Subject: [PATCH] oops forgot mask scaling code --- comfy_api_nodes/nodes_api.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/comfy_api_nodes/nodes_api.py b/comfy_api_nodes/nodes_api.py index 1910e8f74..92f4a0c87 100644 --- a/comfy_api_nodes/nodes_api.py +++ b/comfy_api_nodes/nodes_api.py @@ -365,7 +365,10 @@ class OpenAIGPTImage1(ComfyNodeABC): batch, height, width = mask.shape rgba_mask = torch.zeros(height, width, 4, device="cpu") rgba_mask[:,:,3] = (1-mask.squeeze().cpu()) - mask_np = (rgba_mask.numpy() * 255).astype(np.uint8) + + scaled_mask = downscale_input(rgba_mask.unsqueeze(0)).squeeze() + + mask_np = (scaled_mask.numpy() * 255).astype(np.uint8) mask_img = Image.fromarray(mask_np) mask_img_byte_arr = io.BytesIO() mask_img.save(mask_img_byte_arr, format='PNG')