diff --git a/denoising_diffusion_pytorch/denoising_diffusion_pytorch.py b/denoising_diffusion_pytorch/denoising_diffusion_pytorch.py index 708e09e..7f586f1 100644 --- a/denoising_diffusion_pytorch/denoising_diffusion_pytorch.py +++ b/denoising_diffusion_pytorch/denoising_diffusion_pytorch.py @@ -431,7 +431,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)