diff --git a/audio_diffusion_pytorch/model.py b/audio_diffusion_pytorch/model.py
index a5ac7af..890bb48 100644
--- 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
index 07b1641..9f9a047 100644
--- 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",