From eafb136214955c50798758ed3fb81dd960a626f0 Mon Sep 17 00:00:00 2001
From: Phil Wang <lucidrains@gmail.com>
Date: Tue, 26 Apr 2022 11:40:45 -0700
Subject: [PATCH] suppress a warning

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

diff --git a/dalle2_pytorch/dalle2_pytorch.py b/dalle2_pytorch/dalle2_pytorch.py
index 612dd0f4..f3dbdb5a 100644
--- a/dalle2_pytorch/dalle2_pytorch.py
+++ b/dalle2_pytorch/dalle2_pytorch.py
@@ -84,7 +84,7 @@ def resize_image_to(t, image_size, mode = 'bilinear'): # take a look at https://
     if orig_image_size == shape:
         return t
 
-    return F.interpolate(t, size = shape, mode = mode)
+    return F.interpolate(t, size = shape, mode = mode, align_corners = False)
 
 # classifier free guidance functions
 
diff --git a/setup.py b/setup.py
index 10da6b36..24a42a2d 100644
--- a/setup.py
+++ b/setup.py
@@ -10,7 +10,7 @@
       'dream = dalle2_pytorch.cli:dream'
     ],
   },
-  version = '0.0.50',
+  version = '0.0.51',
   license='MIT',
   description = 'DALL-E 2',
   author = 'Phil Wang',