From edc6d5136a5edddfc7a4960bdb64365c7282a451 Mon Sep 17 00:00:00 2001 From: wassname Date: Thu, 22 Dec 2022 11:00:56 +0800 Subject: [PATCH] Update denoising_diffusion_pytorch_1d.py --- denoising_diffusion_pytorch/denoising_diffusion_pytorch_1d.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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