Merge 6447afa4c2 into 1dae34b0af

fix a bug
the original code don't register the downsamper module into the parent Module.
2026-04-08 06:28:48 +00:00 · 2026-04-01 13:54:33 +00:00 · 2023-11-24 20:19:04 +08:00
1 changed files with 1 additions and 1 deletions
--- a/library/original_unet.py
+++ b/library/original_unet.py
@@ -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
Author	SHA1	Message	Date
Hu Ye	808d9f1baf	Merge `6447afa4c2` into `1dae34b0af`	2026-04-01 13:54:33 +00:00
Hu Ye	6447afa4c2	fix a bug the original code don't register the downsamper module into the parent Module.	2023-11-24 20:19:04 +08:00