fix: remove print statement

archinetai · Dec 14, 2022 · 432557a · 432557a
1 parent 7fa0ad2
commit 432557a
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/audio_diffusion_pytorch/model.py b/audio_diffusion_pytorch/model.py
@@ -120,7 +120,6 @@ def forward(  # type: ignore
         self, x: Tensor, with_info: bool = False, **kwargs
     ) -> Union[Tensor, Tuple[Tensor, Any]]:
         latent, info = self.encode(x, with_info=True)
-        print(latent.shape)
         loss = super().forward(x, channels_list=[latent], **kwargs)
         return (loss, info) if with_info else loss
 

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