fix logits in classifier free guidance example, thanks to @youngwoo-yoon

Author: lucidrains

denoising_diffusion_pytorch/classifier_free_guidance.py

@@ -377,7 +377,7 @@ def forward_with_cond_scale(
         cond_scale = 1.,
         **kwargs
     ):
-        logits = self.forward(*args, **kwargs)
+        logits = self.forward(*args, cond_drop_prob = 0., **kwargs)
 
         if cond_scale == 1:
             return logits

denoising_diffusion_pytorch/version.py

@@ -1 +1 @@
-__version__ = '1.5.3'
+__version__ = '1.5.4'