This commit is contained in:
Hu Ye
2026-04-04 21:41:36 +00:00
committed by GitHub

View File

@@ -508,7 +508,7 @@ class DownBlock2D(nn.Module):
self.resnets = nn.ModuleList(resnets)
if add_downsample:
self.downsamplers = [Downsample2D(out_channels, out_channels=out_channels)]
self.downsamplers = nn.ModuleList([Downsample2D(out_channels, out_channels=out_channels)])
else:
self.downsamplers = None