diff --git a/comfy/ldm/modules/diffusionmodules/model.py b/comfy/ldm/modules/diffusionmodules/model.py index 74322c27..ed1e8821 100644 --- a/comfy/ldm/modules/diffusionmodules/model.py +++ b/comfy/ldm/modules/diffusionmodules/model.py @@ -91,7 +91,7 @@ class Upsample(nn.Module): def forward(self, x): scale_factor = self.scale_factor - if not isinstance(scale_factor, tuple): + if isinstance(scale_factor, (int, float)): scale_factor = (scale_factor,) * (x.ndim - 2) if x.ndim == 5 and scale_factor[0] > 1.0: @@ -109,7 +109,7 @@ class Upsample(nn.Module): else: x = a else: - x = interpolate_up(x, self.scale_factor) + x = interpolate_up(x, scale_factor) if self.with_conv: x = self.conv(x) return x