address #346

lucidrains · Oct 9, 2024 · 1d9d8df · 1d9d8df
1 parent 403f9e6
commit 1d9d8df
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/denoising_diffusion_pytorch/classifier_free_guidance.py b/denoising_diffusion_pytorch/classifier_free_guidance.py
@@ -646,7 +646,7 @@ def model_predictions(self, x, t, classes, cond_scale = 6., rescaled_phi = 0.7,
         if self.objective == 'pred_noise':
             pred_noise = model_output if not self.use_cfg_plus_plus else model_output_null
 
-            x_start = self.predict_start_from_noise(x, t, pred_noise)
+            x_start = self.predict_start_from_noise(x, t, model_output)
             x_start = maybe_clip(x_start)
 
         elif self.objective == 'pred_x0':

diff --git a/denoising_diffusion_pytorch/version.py b/denoising_diffusion_pytorch/version.py
@@ -1 +1 @@
-__version__ = '2.1.0'
+__version__ = '2.1.1'