nodes/pixaroma-ComfyUI-Pixaroma
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix(inpaint): crop_info mask stays float32 on fp16 pipelines

Browse Source 
The carried full-frame mask followed image.dtype, so on an fp16 image it came
back float16 while out_mask was float32. Pin it to float32 for consistency.

Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
This commit is contained in:
pixaroma
2026-06-16 14:51:30 +03:00
co-authored by Claude Opus 4.8
parent cad1586ec2
commit 2ef7cd6cde
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
nodes/_inpaint_helpers.py
+3 -1
View File
@@ -402,7 +402,9 @@ def apply_inpaint_crop(image, mask, p):
mout = gaussian_blur_np(mout, p["mask_blur"])
out_mask = torch.from_numpy(np.clip(mout, 0, 1)[None, ...].astype(np.float32)).to(image.device)
full_mask = torch.from_numpy(softm[None, ...].astype(np.float32)).to(image.device, image.dtype)
# keep the carried mask float32 (NOT image.dtype) so it matches out_mask and a
# strict downstream mask op never sees a float16 mask on an fp16 image pipeline.
full_mask = torch.from_numpy(softm[None, ...].astype(np.float32)).to(image.device)
crop_info = {
"image": image, "mask": full_mask,
"x": rx, "y": ry, "w": rw, "h": rh,