diff --git a/setup.py b/setup.py
index 0e77444..393a128 100644
--- a/setup.py
+++ b/setup.py
@@ -3,7 +3,7 @@
 setup(
     name="audio-diffusion-pytorch",
     packages=find_packages(exclude=[]),
-    version="0.0.1",
+    version="0.0.2",
     license="MIT",
     description="Audio Diffusion - PyTorch",
     long_description_content_type="text/markdown",