RomanNum3ral
/
ComfyUI
mirror of https://github.com/Comfy-Org/ComfyUI.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Merge 012d0559cd into 37ac9ff44f

This commit is contained in:
seesaw game 2026-08-15 20:51:43 +02:00 committed by GitHub
parent 37ac9ff44f 012d0559cd
commit 8f6df5a137
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 19 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
comfy_extras/frame_interpolation_models/film_net.py
View File

@ -33,7 +33,17 @@ def _warp_core(image, flow, grid_x, grid_y):
dx = flow[:, 0].float() / (W * 0.5)
dy = flow[:, 1].float() / (H * 0.5)
grid = torch.stack([grid_x[None, None, :] + dx, grid_y[None, :, None] + dy], dim=3)
return F.grid_sample(image.float(), grid, mode="bilinear", padding_mode="border", align_corners=False).to(dtype)
padding_mode = "border"
if image.device.type == "mps":
# MPS does not implement "border" padding. With align_corners=False the valid
# sampling range is +-(1 - 1/size) per axis (grid[..., 0] is x -> width,
# grid[..., 1] is y -> height), so clamping there makes "zeros" produce
# results identical to "border".
bx = 1.0 - 1.0 / image.shape[-1]
by = 1.0 - 1.0 / image.shape[-2]
grid = torch.stack([grid[..., 0].clamp(-bx, bx), grid[..., 1].clamp(-by, by)], dim=-1)
padding_mode = "zeros"
return F.grid_sample(image.float(), grid, mode="bilinear", padding_mode=padding_mode, align_corners=False).to(dtype)
def build_image_pyramid(image, pyramid_levels):

9
comfy_extras/frame_interpolation_models/ifnet.py
View File

@ -12,7 +12,14 @@ def _warp(img, flow, warp_grids):
base_grid, flow_div = warp_grids[(H, W)]
flow_norm = torch.cat([flow[:, 0:1] / flow_div[0], flow[:, 1:2] / flow_div[1]], 1).float()
grid = (base_grid.expand(B, -1, -1, -1) + flow_norm).permute(0, 2, 3, 1)
return F.grid_sample(img.float(), grid, mode="bilinear", padding_mode="border", align_corners=True).to(img.dtype)
padding_mode = "border"
if img.device.type == "mps":
# MPS does not implement "border" padding. With align_corners=True the valid
# sampling range is exactly [-1, 1], so clamping the grid to it makes "zeros"
# produce results identical to "border".
grid = grid.clamp(-1.0, 1.0)
padding_mode = "zeros"
return F.grid_sample(img.float(), grid, mode="bilinear", padding_mode=padding_mode, align_corners=True).to(img.dtype)
class Head(nn.Module):