Merge 6447afa4c2 into 51435f1718

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-04 21:41:36 +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	7142e88e24	Merge `6447afa4c2` into `51435f1718`	2026-04-04 21:41:36 +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