Update denoising_diffusion_pytorch_1d.py

2026-09-09 11:21:11 +08:00 · 2022-12-22 11:00:56 +08:00
parent 207d23af86
commit edc6d5136a
1 changed files with 1 additions and 1 deletions
@@ -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