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

2026-02-14 04:15:40 +01:00 · 2022-07-14 15:28:34 -07:00
2 changed files with 2 additions and 2 deletions
--- a/dalle2_pytorch/dalle2_pytorch.py
+++ b/dalle2_pytorch/dalle2_pytorch.py
@@ -1926,7 +1926,7 @@ class Unet(nn.Module):
                hiddens.append(x)

            x = attn(x)
-            hiddens.append(x)
+            hiddens.append(x.contiguous())

            if exists(post_downsample):
                x = post_downsample(x)
--- a/dalle2_pytorch/version.py
+++ b/dalle2_pytorch/version.py
@@ -1 +1 @@
-__version__ = '0.24.1'
+__version__ = '0.24.2'