fix: v-sampler output

archinetai · Oct 18, 2022 · 7f6151e · 7f6151e
1 parent 954f9be
commit 7f6151e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/audio_diffusion_pytorch/diffusion.py b/audio_diffusion_pytorch/diffusion.py
@@ -388,7 +388,7 @@ def forward(
                 alpha, beta = self.get_alpha_beta(sigmas[i + 1].item())
                 x = x_pred * alpha + x_eps * beta
 
-        return x
+        return x_pred
 
 
 class KarrasSampler(Sampler):

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