diff --git a/src/diffusers/models/unets/unet_1d_blocks.py b/src/diffusers/models/unets/unet_1d_blocks.py index 3c7c1cbecee9..8fc27e94c474 100644 --- a/src/diffusers/models/unets/unet_1d_blocks.py +++ b/src/diffusers/models/unets/unet_1d_blocks.py @@ -200,7 +200,7 @@ def __init__( 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: