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

takes care of a grad strides error at https://github.com/lucidrains/DALLE2-pytorch/issues/196 thanks to @YUHANG-Ma

Browse Source
This commit is contained in:
Phil Wang
2022-07-14 15:28:34 -07:00
parent 1662bbf226
commit a58a370d75
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
dalle2_pytorch/dalle2_pytorch.py
View File

@@ -1926,7 +1926,7 @@ class Unet(nn.Module):
hiddens.append(x) hiddens.append(x)
x = attn(x) x = attn(x)
hiddens.append(x) hiddens.append(x.contiguous())
if exists(post_downsample): if exists(post_downsample):
x = post_downsample(x) x = post_downsample(x)

2
dalle2_pytorch/version.py
View File

@@ -1 +1 @@
__version__ = '0.24.1' __version__ = '0.24.2'