self.upsample = Upsample1D (#8580)

Making self.upsample actually be Upsample1D

self.upsample = Upsample1D (#8580)
Making self.upsample actually be Upsample1D
4408047a · kkj15dk · GitHub · 34fab8b5 · 4408047a
Unverified Commit 4408047a authored Jun 18, 2024 by kkj15dk Committed by GitHub Jun 18, 2024
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/diffusers/models/unets/unet_1d_blocks.py src/diffusers/models/unets/unet_1d_blocks.py +1 -1

No files found.
--- 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: