wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-07-07 07:01:46 +08:00
Code Issues Actions 14 Packages Projects Releases Wiki Activity

Add docstring for MoGe interpolation helper

Browse Source
This commit is contained in:
holys519 2026-06-26 10:39:06 +09:00
parent ee847d4ae0
commit 2e9d6241ea
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
comfy/ldm/moge/modules.py
View File

@ -30,6 +30,7 @@ def _concat_view_plane_uv(x: torch.Tensor, aspect_ratio: float) -> torch.Tensor:
def _interpolate_antialias_safe(input: torch.Tensor, *args, **kwargs) -> torch.Tensor: def _interpolate_antialias_safe(input: torch.Tensor, *args, **kwargs) -> torch.Tensor:
"""Run antialiased interpolation in fp32 when half precision is unsupported."""
if kwargs.get("antialias") and input.dtype in (torch.float16, torch.bfloat16): if kwargs.get("antialias") and input.dtype in (torch.float16, torch.bfloat16):
return F.interpolate(input.float(), *args, **kwargs).to(dtype=input.dtype) return F.interpolate(input.float(), *args, **kwargs).to(dtype=input.dtype)
return F.interpolate(input, *args, **kwargs) return F.interpolate(input, *args, **kwargs)