address #346

Author: lucidrains

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':

denoising_diffusion_pytorch/version.py

@@ -1 +1 @@
-__version__ = '2.1.0'
+__version__ = '2.1.1'