aljaz/DALLE2-pytorch
1
1
Fork 0
mirror of https://github.com/lucidrains/DALLE2-pytorch.git synced 2025-12-19 09:44:19 +01:00
Code Issues Packages Projects Releases Wiki Activity

one more residual, after seeing good results on unconditional generation locally

Browse Source
This commit is contained in:
Phil Wang
2022-06-16 11:18:02 -07:00
parent e6bb75e5ab
commit 6651eafa93
2 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
dalle2_pytorch/dalle2_pytorch.py
View File

@@ -1488,7 +1488,7 @@ class Unet(nn.Module):
])) ]))
self.final_conv = nn.Sequential( self.final_conv = nn.Sequential(
ResnetBlock(dim, dim, groups = resnet_groups[0]), ResnetBlock(dim * 2, dim, groups = resnet_groups[0]),
nn.Conv2d(dim, self.channels_out, 1) nn.Conv2d(dim, self.channels_out, 1)
) )
@@ -1560,6 +1560,7 @@ class Unet(nn.Module):
# initial convolution # initial convolution
x = self.init_conv(x) x = self.init_conv(x)
r = x.clone() # final residual
# time conditioning # time conditioning
@@ -1689,6 +1690,7 @@ class Unet(nn.Module):
x = upsample(x) x = upsample(x)
x = torch.cat((x, r), dim = 1)
return self.final_conv(x) return self.final_conv(x)
class LowresConditioner(nn.Module): class LowresConditioner(nn.Module):

2
dalle2_pytorch/version.py
View File

@@ -1 +1 @@
__version__ = '0.9.1' __version__ = '0.9.2'