From 7658101c2ea85e6841c1fd67abe1f4e8ffd5c7f2 Mon Sep 17 00:00:00 2001
From: lucidrains <lucidrains@gmail.com>
Date: Wed, 27 Sep 2023 07:39:30 -0700
Subject: [PATCH] address
 https://github.com/lucidrains/denoising-diffusion-pytorch/issues/256

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

diff --git a/denoising_diffusion_pytorch/classifier_free_guidance.py b/denoising_diffusion_pytorch/classifier_free_guidance.py
index 1253a61be..1187aa161 100644
--- a/denoising_diffusion_pytorch/classifier_free_guidance.py
+++ b/denoising_diffusion_pytorch/classifier_free_guidance.py
@@ -671,7 +671,7 @@ def ddim_sample(self, classes, shape, cond_scale = 6., rescaled_phi = 0.7, clip_
 
         for time, time_next in tqdm(time_pairs, desc = 'sampling loop time step'):
             time_cond = torch.full((batch,), time, device=device, dtype=torch.long)
-            pred_noise, x_start, *_ = self.model_predictions(img, time_cond, classes, cond_scale = cond_scale, clip_x_start = clip_denoised)
+            pred_noise, x_start, *_ = self.model_predictions(img, time_cond, classes, cond_scale = cond_scale, rescaled_phi = rescaled_phi, clip_x_start = clip_denoised)
 
             if time_next < 0:
                 img = x_start
diff --git a/denoising_diffusion_pytorch/version.py b/denoising_diffusion_pytorch/version.py
index 4c6f4f001..520dbc955 100644
--- a/denoising_diffusion_pytorch/version.py
+++ b/denoising_diffusion_pytorch/version.py
@@ -1 +1 @@
-__version__ = '1.8.12'
+__version__ = '1.8.14'