From aafed3f8dd042bfe786f6c3e902c5cdb5de1fb08 Mon Sep 17 00:00:00 2001 From: Kaiwen Sheng Date: Mon, 16 Dec 2024 15:25:16 -0800 Subject: [PATCH] fix downsample bug in MidResTemporalBlock1D (#10250) --- src/diffusers/models/unets/unet_1d_blocks.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/diffusers/models/unets/unet_1d_blocks.py b/src/diffusers/models/unets/unet_1d_blocks.py index 8fc27e94c474..f08e6070845e 100644 --- a/src/diffusers/models/unets/unet_1d_blocks.py +++ b/src/diffusers/models/unets/unet_1d_blocks.py @@ -217,7 +217,7 @@ def forward(self, hidden_states: torch.Tensor, temb: torch.Tensor) -> torch.Tens if self.upsample: hidden_states = self.upsample(hidden_states) if self.downsample: - self.downsample = self.downsample(hidden_states) + hidden_states = self.downsample(hidden_states) return hidden_states