xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-09-09 18:27:03 +08:00
Code Issues Packages Projects Releases Wiki Activity

oops forgot mask scaling code

Browse Source
This commit is contained in:
thot-experiment 2025-04-23 11:50:26 -07:00
parent 6fb5bb1431
commit 18fba2a3c0
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
comfy_api_nodes/nodes_api.py
View File

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