From 3dae43fa0ef2c5a29df9c275b1ebeb1a65c3bac5 Mon Sep 17 00:00:00 2001
From: Phil Wang <lucidrains@gmail.com>
Date: Sat, 9 Jul 2022 19:01:37 -0700
Subject: [PATCH] fix misnamed variable, thanks to @nousr

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

diff --git a/dalle2_pytorch/dalle2_pytorch.py b/dalle2_pytorch/dalle2_pytorch.py
index ed4ebe4..bf1360d 100644
--- a/dalle2_pytorch/dalle2_pytorch.py
+++ b/dalle2_pytorch/dalle2_pytorch.py
@@ -1061,9 +1061,9 @@ class DiffusionPrior(nn.Module):
             c2 = ((1 - alpha_next) - torch.square(c1)).sqrt()
             noise = torch.randn_like(image_embed) if time_next > 0 else 0.
 
-            img = x_start * alpha_next.sqrt() + \
-                  c1 * noise + \
-                  c2 * pred_noise
+            image_embed = x_start * alpha_next.sqrt() + \
+                          c1 * noise + \
+                          c2 * pred_noise
 
         return image_embed
 
diff --git a/dalle2_pytorch/version.py b/dalle2_pytorch/version.py
index 2c49b72..cd5705a 100644
--- a/dalle2_pytorch/version.py
+++ b/dalle2_pytorch/version.py
@@ -1 +1 @@
-__version__ = '0.19.4'
+__version__ = '0.19.5'