From 067ac323da8c0e2314f954dd94308cfbc2015f9b Mon Sep 17 00:00:00 2001
From: Phil Wang <lucidrains@gmail.com>
Date: Wed, 23 Nov 2022 08:41:18 -0800
Subject: [PATCH] address
 https://github.com/lucidrains/DALLE2-pytorch/issues/266

---
 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 bc71e91..c3b353d 100644
--- a/dalle2_pytorch/dalle2_pytorch.py
+++ b/dalle2_pytorch/dalle2_pytorch.py
@@ -1320,7 +1320,7 @@ class DiffusionPrior(nn.Module):
             elif self.predict_x_start:
                 x_start = pred
             else:
-                x_start = self.noise_scheduler.predict_start_from_noise(image_embed, t = time_cond, noise = pred_noise)
+                x_start = self.noise_scheduler.predict_start_from_noise(image_embed, t = time_cond, noise = pred)
 
             # clip x0 before maybe predicting noise
 
diff --git a/dalle2_pytorch/version.py b/dalle2_pytorch/version.py
index 522ba08..6c371de 100644
--- a/dalle2_pytorch/version.py
+++ b/dalle2_pytorch/version.py
@@ -1 +1 @@
-__version__ = '1.11.1'
+__version__ = '1.11.2'