diff --git a/denoising_diffusion_pytorch/denoising_diffusion_pytorch_1d.py b/denoising_diffusion_pytorch/denoising_diffusion_pytorch_1d.py index 327f34f..9ceaadf 100644 --- a/denoising_diffusion_pytorch/denoising_diffusion_pytorch_1d.py +++ b/denoising_diffusion_pytorch/denoising_diffusion_pytorch_1d.py @@ -414,7 +414,7 @@ class GaussianDiffusion1D(nn.Module): beta_schedule = 'cosine', p2_loss_weight_gamma = 0., p2_loss_weight_k = 1, - ddim_sampling_eta = 1. + ddim_sampling_eta = 0. ): super().__init__() self.model = model