fix: add linear schedule in init

archinetai · Oct 17, 2022 · 954f9be · 954f9be
1 parent 6c9b5d4
commit 954f9be
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/audio_diffusion_pytorch/__init__.py b/audio_diffusion_pytorch/__init__.py
@@ -8,6 +8,7 @@
     KarrasSampler,
     KarrasSchedule,
     KDiffusion,
+    LinearSchedule,
     LogNormalDistribution,
     Sampler,
     Schedule,

diff --git a/setup.py b/setup.py
@@ -3,7 +3,7 @@
 setup(
     name="audio-diffusion-pytorch",
     packages=find_packages(exclude=[]),
-    version="0.0.67",
+    version="0.0.68",
     license="MIT",
     description="Audio Diffusion - PyTorch",
     long_description_content_type="text/markdown",