VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-08 06:28:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix a bug

Browse Source 
the original code don't register the downsamper module into the parent Module.
This commit is contained in:
Hu Ye
2023-11-24 20:19:04 +08:00
committed by GitHub
parent 95ae56bd22
commit 6447afa4c2
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
library/original_unet.py
View File

@@ -485,7 +485,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