Skip to content

GitLab

"...models/git@developer.sourcefind.cn:OpenDAS/vision.git" did not exist on "b3cdec1fd808b287e4e0d31f989488c335df1812"
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): ...@@ -200,7 +200,7 @@ class MidResTemporalBlock1D(nn.Module):
self.upsample = None self.upsample = None
if add_upsample: if add_upsample:
self.upsample = Downsample1D(out_channels, use_conv=True) self.upsample = Upsample1D(out_channels, use_conv=True)
self.downsample = None self.downsample = None
if add_downsample: 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