Update denoising_diffusion_pytorch_1d.py

wassname · web-flow · commit edc6d5136a5e · 2022-12-22T11:00:56.000+08:00
diff --git a/denoising_diffusion_pytorch/denoising_diffusion_pytorch_1d.py b/denoising_diffusion_pytorch/denoising_diffusion_pytorch_1d.py
@@ -414,7 +414,7 @@ def __init__(
         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
