Downsample in the MidResTemporalBlock1D

[holmosaint]

Describe the bug

In decoding/model/autoencoders/unets/unet_1d_blocks.py:212-222

 def forward(self, hidden_states: torch.Tensor, temb: torch.Tensor) -> torch.Tensor:
 hidden_states = self.resnets[0](hidden_states, temb)
 for resnet in self.resnets[1:]:
 hidden_states = resnet(hidden_states, temb)

 if self.upsample:
 hidden_states = self.upsample(hidden_states)
 if self.downsample:
 self.downsample = self.downsample(hidden_states)

 return hidden_states

Looks like it should be

hidden_states = self.downsample(hidden_states)

Reproduction

N/A

Logs

No response

System Info

N/A

Who can help?

No response

[a-r-r-o-w]

@holmosaint Wow, not sure how that slipped by. Thanks for the find @holmosaint! Would you like to open a PR with the suggested fix?

[holmosaint]

Yeah, totally, will do it shortly today and then close the issue afterwards.