From ed620228f7685e9576fdad365ccb791a8f9ad97c Mon Sep 17 00:00:00 2001
From: Phil Wang <lucidrains@gmail.com>
Date: Fri, 23 Dec 2022 13:13:22 -0800
Subject: [PATCH] default to sigmoid noise schedule, seems to be a lot better

---
 denoising_diffusion_pytorch/denoising_diffusion_pytorch_1d.py | 2 +-
 denoising_diffusion_pytorch/version.py                        | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/denoising_diffusion_pytorch/denoising_diffusion_pytorch_1d.py b/denoising_diffusion_pytorch/denoising_diffusion_pytorch_1d.py
index 9ceaadf14..4ca735bc2 100644
--- a/denoising_diffusion_pytorch/denoising_diffusion_pytorch_1d.py
+++ b/denoising_diffusion_pytorch/denoising_diffusion_pytorch_1d.py
@@ -411,7 +411,7 @@ def __init__(
         sampling_timesteps = None,
         loss_type = 'l1',
         objective = 'pred_noise',
-        beta_schedule = 'cosine',
+        beta_schedule = 'sigmoid',
         p2_loss_weight_gamma = 0.,
         p2_loss_weight_k = 1,
         ddim_sampling_eta = 0.
diff --git a/denoising_diffusion_pytorch/version.py b/denoising_diffusion_pytorch/version.py
index 124e46203..020ed73d7 100644
--- a/denoising_diffusion_pytorch/version.py
+++ b/denoising_diffusion_pytorch/version.py
@@ -1 +1 @@
-__version__ = '0.1.7'
+__version__ = '0.2.2'