andreas/ComfyUI
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Remove workaround for old pytorch. (#12480)

Browse Source
This commit is contained in:
comfyanonymous
2026-02-15 17:43:53 -08:00
committed by GitHub
parent c0370044cd
commit 88e6370527
1 changed files with 1 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files
comfy/ldm/modules/diffusionmodules/model.py
-12
View File
@@ -102,19 +102,7 @@ class VideoConv3d(nn.Module):
return self.conv(x) return self.conv(x)
def interpolate_up(x, scale_factor): def interpolate_up(x, scale_factor):
try:
return torch.nn.functional.interpolate(x, scale_factor=scale_factor, mode="nearest") return torch.nn.functional.interpolate(x, scale_factor=scale_factor, mode="nearest")
except: #operation not implemented for bf16
orig_shape = list(x.shape)
out_shape = orig_shape[:2]
for i in range(len(orig_shape) - 2):
out_shape.append(round(orig_shape[i + 2] * scale_factor[i]))
out = torch.empty(out_shape, dtype=x.dtype, layout=x.layout, device=x.device)
split = 8
l = out.shape[1] // split
for i in range(0, out.shape[1], l):
out[:,i:i+l] = torch.nn.functional.interpolate(x[:,i:i+l].to(torch.float32), scale_factor=scale_factor, mode="nearest").to(x.dtype)
return out
class Upsample(nn.Module): class Upsample(nn.Module):
def __init__(self, in_channels, with_conv, conv_op=ops.Conv2d, scale_factor=2.0): def __init__(self, in_channels, with_conv, conv_op=ops.Conv2d, scale_factor=2.0):