Merge 6447afa4c2 into a21b6a917e

2026-04-16 17:02:45 +00:00 · 2025-05-12 19:32:36 +02:00
parent a21b6a917e 6447afa4c2
commit 925f2f7618
1 changed files with 1 additions and 1 deletions
--- a/library/original_unet.py
+++ b/library/original_unet.py
@@ -506,7 +506,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