Merge pull request #139 from wassname/patch-1

eta=0

Author: lucidrains

denoising_diffusion_pytorch/denoising_diffusion_pytorch.py

@@ -431,7 +431,7 @@ def __init__(
         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)

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