diff --git a/denoising_diffusion_pytorch/version.py b/denoising_diffusion_pytorch/version.py
index afced1472..3f3907994 100644
--- a/denoising_diffusion_pytorch/version.py
+++ b/denoising_diffusion_pytorch/version.py
@@ -1 +1 @@
-__version__ = '2.0.0'
+__version__ = '2.0.1'