From a58a370d7562d0b072697d4b4597097c294376c6 Mon Sep 17 00:00:00 2001
From: Phil Wang <lucidrains@gmail.com>
Date: Thu, 14 Jul 2022 15:28:34 -0700
Subject: [PATCH] takes care of a grad strides error at
 https://github.com/lucidrains/DALLE2-pytorch/issues/196 thanks to @YUHANG-Ma

---
 dalle2_pytorch/dalle2_pytorch.py | 2 +-
 dalle2_pytorch/version.py        | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/dalle2_pytorch/dalle2_pytorch.py b/dalle2_pytorch/dalle2_pytorch.py
index 862440e..4626080 100644
--- 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)
diff --git a/dalle2_pytorch/version.py b/dalle2_pytorch/version.py
index 7fe1493..d1d123f 100644
--- a/dalle2_pytorch/version.py
+++ b/dalle2_pytorch/version.py
@@ -1 +1 @@
-__version__ = '0.24.1'
+__version__ = '0.24.2'