298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-01-11 10:25:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix regression in VAE code on old pytorch versions.

Browse Source
This commit is contained in:
comfyanonymous 2024-12-18 03:08:28 -05:00
parent 79badea452
commit 4c5c4ddeda
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
comfy/ldm/modules/diffusionmodules/model.py
View File

@ -91,7 +91,7 @@ class Upsample(nn.Module):
def forward(self, x): def forward(self, x):
scale_factor = self.scale_factor scale_factor = self.scale_factor
if not isinstance(scale_factor, tuple): if isinstance(scale_factor, (int, float)):
scale_factor = (scale_factor,) * (x.ndim - 2) scale_factor = (scale_factor,) * (x.ndim - 2)
if x.ndim == 5 and scale_factor[0] > 1.0: if x.ndim == 5 and scale_factor[0] > 1.0:
@ -109,7 +109,7 @@ class Upsample(nn.Module):
else: else:
x = a x = a
else: else:
x = interpolate_up(x, self.scale_factor) x = interpolate_up(x, scale_factor)
if self.with_conv: if self.with_conv:
x = self.conv(x) x = self.conv(x)
return x return x