takes care of a grad strides error at #196 thanks to @YUHANG-Ma

Author: lucidrains

dalle2_pytorch/dalle2_pytorch.py

@@ -1926,7 +1926,7 @@ def forward(
                 hiddens.append(x)
 
             x = attn(x)
-            hiddens.append(x)
+            hiddens.append(x.contiguous())
 
             if exists(post_downsample):
                 x = post_downsample(x)

dalle2_pytorch/version.py

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