Skip to content

GitLab

Unverified Commit aafed3f8 authored by Kaiwen Sheng's avatar Kaiwen Sheng Committed by GitHub
Browse files

fix downsample bug in MidResTemporalBlock1D (#10250)

parent 5ed761a6
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/diffusers/models/unets/unet_1d_blocks.py
View file @ aafed3f8
...@@ -217,7 +217,7 @@ class MidResTemporalBlock1D(nn.Module): ...@@ -217,7 +217,7 @@ class MidResTemporalBlock1D(nn.Module):
if self.upsample: if self.upsample:
hidden_states = self.upsample(hidden_states) hidden_states = self.upsample(hidden_states)
if self.downsample: if self.downsample:
self.downsample = self.downsample(hidden_states) hidden_states = self.downsample(hidden_states)
return hidden_states return hidden_states
......
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