Skip to content

Commit 4408047

Browse files
authored
self.upsample = Upsample1D (#8580)
Making self.upsample actually be Upsample1D
1 parent 34fab8b commit 4408047

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/diffusers/models/unets/unet_1d_blocks.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -200,7 +200,7 @@ def __init__(
200200

201201
self.upsample = None
202202
if add_upsample:
203-
self.upsample = Downsample1D(out_channels, use_conv=True)
203+
self.upsample = Upsample1D(out_channels, use_conv=True)
204204

205205
self.downsample = None
206206
if add_downsample:

0 commit comments

Comments
 (0)