AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

self.upsample = Upsample1D (#8580)

Browse Source 
Making self.upsample actually be Upsample1D
This commit is contained in:
kkj15dk
2024-06-18 14:34:07 -07:00
committed by GitHub
parent 34fab8b511
commit 4408047ac5
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/diffusers/models/unets/unet_1d_blocks.py
View File

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