Merge 6447afa4c2 into b2abe873a5

Merge pull request #2283 from kozistr/deps/pytorch-optimizer
Bump `pytorch-optimizer` into 3.10.0
2026-04-18 01:30:02 +00:00 · 2026-03-29 14:31:42 +00:00 · 2026-03-19 09:18:06 +09:00 · 2026-03-02 14:39:48 +09:00 · 2023-11-24 20:19:04 +08:00
2 changed files with 2 additions and 2 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

--- a/requirements.txt
+++ b/requirements.txt
@@ -9,7 +9,7 @@ einops==0.7.0
 bitsandbytes
 lion-pytorch==0.2.3
 schedulefree==1.4
-pytorch-optimizer==3.9.0
+pytorch-optimizer==3.10.0
 prodigy-plus-schedule-free==1.9.2
 prodigyopt==1.1.2
 tensorboard
Author	SHA1	Message	Date
Hu Ye	c97a787a1e	Merge `6447afa4c2` into `b2abe873a5`	2026-03-29 14:31:42 +00:00
Kohya S.	b2abe873a5	Merge pull request #2283 from kozistr/deps/pytorch-optimizer Bump `pytorch-optimizer` into 3.10.0	2026-03-19 09:18:06 +09:00
kozistr	1bd0b0faf1	build(deps): bump pytorch-optimizer into 3.10.0	2026-03-02 14:39:48 +09: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