address #256

Author: lucidrains

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

denoising_diffusion_pytorch/version.py

@@ -1 +1 @@
-__version__ = '1.8.12'
+__version__ = '1.8.14'