From 4408047ac557a10244413f5f650c2da7106567a1 Mon Sep 17 00:00:00 2001
From: kkj15dk <76531122+kkj15dk@users.noreply.github.com>
Date: Tue, 18 Jun 2024 14:34:07 -0700
Subject: [PATCH] self.upsample = Upsample1D (#8580)

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 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: