From ebaa1a4d62e3227a9a6a816c0ccb90e9e6f12522 Mon Sep 17 00:00:00 2001 From: Kasper Krunderup Jakobsen Date: Sat, 15 Jun 2024 22:33:48 -0700 Subject: [PATCH] Making self.upsample actually be Upsample1D --- src/diffusers/models/unets/unet_1d_blocks.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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: