Downsample in the MidResTemporalBlock1D #10247
Description
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_statesLooks 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
