Skip to content

Commit d5cb380

Browse files
holmosaintsayakpaul
authored andcommitted
fix downsample bug in MidResTemporalBlock1D (#10250)
1 parent c3765a9 commit d5cb380

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/diffusers/models/unets/unet_1d_blocks.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -217,7 +217,7 @@ def forward(self, hidden_states: torch.Tensor, temb: torch.Tensor) -> torch.Tens
217217
if self.upsample:
218218
hidden_states = self.upsample(hidden_states)
219219
if self.downsample:
220-
self.downsample = self.downsample(hidden_states)
220+
hidden_states = self.downsample(hidden_states)
221221

222222
return hidden_states
223223

0 commit comments

Comments
 (0)