Skip to content

GitLab

Unverified Commit 4408047a authored by kkj15dk's avatar kkj15dk Committed by GitHub
Browse files

self.upsample = Upsample1D (#8580) 

Making self.upsample actually be Upsample1D
parent 34fab8b5
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/diffusers/models/unets/unet_1d_blocks.py
View file @ 4408047a
......@@ -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:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment