wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-12-17 01:52:59 +08:00
Code Issues Actions 13 Packages Projects Releases Wiki Activity

Fix VAEEncodeForInpaint to support WAN VAE tuple downscale_ratio

Browse Source 
- Handle WAN VAE downscale_ratio which is a tuple (function, min, max) instead of integer
- Fixes TypeError when using VAEEncodeForInpaint with Qwen Image/WAN VAEs
- Backward compatible with standard VAEs that use integer downscale_ratio
This commit is contained in:
ChrisFab16 2025-12-08 21:25:35 +01:00
parent fd271dedfd
commit 6d82490b88
1 changed files with 22 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
nodes.py
View File

@ -374,14 +374,32 @@ class VAEEncodeForInpaint:
CATEGORY = "latent/inpaint" CATEGORY = "latent/inpaint"
def encode(self, vae, pixels, mask, grow_mask_by=6): def encode(self, vae, pixels, mask, grow_mask_by=6):
x = (pixels.shape[1] // vae.downscale_ratio) * vae.downscale_ratio # Handle WAN VAE downscale_ratio which can be a tuple (function, min, max)
y = (pixels.shape[2] // vae.downscale_ratio) * vae.downscale_ratio if isinstance(vae.downscale_ratio, tuple):
# For WAN VAEs: (lambda function, min_value, max_value)
downscale_func = vae.downscale_ratio[0]
x_downscale = downscale_func(pixels.shape[1])
y_downscale = downscale_func(pixels.shape[2])
x = (pixels.shape[1] // x_downscale) * x_downscale
y = (pixels.shape[2] // y_downscale) * y_downscale
else:
# Standard integer downscale_ratio
x = (pixels.shape[1] // vae.downscale_ratio) * vae.downscale_ratio
y = (pixels.shape[2] // vae.downscale_ratio) * vae.downscale_ratio
mask = torch.nn.functional.interpolate(mask.reshape((-1, 1, mask.shape[-2], mask.shape[-1])), size=(pixels.shape[1], pixels.shape[2]), mode="bilinear") mask = torch.nn.functional.interpolate(mask.reshape((-1, 1, mask.shape[-2], mask.shape[-1])), size=(pixels.shape[1], pixels.shape[2]), mode="bilinear")
pixels = pixels.clone() pixels = pixels.clone()
if pixels.shape[1] != x or pixels.shape[2] != y: if pixels.shape[1] != x or pixels.shape[2] != y:
x_offset = (pixels.shape[1] % vae.downscale_ratio) // 2 if isinstance(vae.downscale_ratio, tuple):
y_offset = (pixels.shape[2] % vae.downscale_ratio) // 2 downscale_func = vae.downscale_ratio[0]
x_downscale = downscale_func(pixels.shape[1])
y_downscale = downscale_func(pixels.shape[2])
x_offset = (pixels.shape[1] % x_downscale) // 2
y_offset = (pixels.shape[2] % y_downscale) // 2
else:
x_offset = (pixels.shape[1] % vae.downscale_ratio) // 2
y_offset = (pixels.shape[2] % vae.downscale_ratio) // 2
pixels = pixels[:,x_offset:x + x_offset, y_offset:y + y_offset,:] pixels = pixels[:,x_offset:x + x_offset, y_offset:y + y_offset,:]
mask = mask[:,:,x_offset:x + x_offset, y_offset:y + y_offset] mask = mask[:,:,x_offset:x + x_offset, y_offset:y + y_offset]