diff --git a/src/diffusers/models/unets/unet_1d_blocks.py b/src/diffusers/models/unets/unet_1d_blocks.py index 3c7c1cbece..8fc27e94c4 100644 --- a/src/diffusers/models/unets/unet_1d_blocks.py +++ b/src/diffusers/models/unets/unet_1d_blocks.py @@ -200,7 +200,7 @@ class MidResTemporalBlock1D(nn.Module): self.upsample = None if add_upsample: - self.upsample = Downsample1D(out_channels, use_conv=True) + self.upsample = Upsample1D(out_channels, use_conv=True) self.downsample = None if add_downsample: