diff --git a/denoising_diffusion_pytorch/classifier_free_guidance.py b/denoising_diffusion_pytorch/classifier_free_guidance.py index cc941fd..28a20ed 100644 --- a/denoising_diffusion_pytorch/classifier_free_guidance.py +++ b/denoising_diffusion_pytorch/classifier_free_guidance.py @@ -488,7 +488,7 @@ class GaussianDiffusion(nn.Module): beta_schedule = 'cosine', p2_loss_weight_gamma = 0., # p2 loss weight, from https://arxiv.org/abs/2204.00227 - 0 is equivalent to weight of 1 across time - 1. is recommended p2_loss_weight_k = 1, - ddim_sampling_eta = 1. + ddim_sampling_eta = 0. ): super().__init__() assert not (type(self) == GaussianDiffusion and model.channels != model.out_dim)