Skip to content

Commit 432557a

Browse files
fix: remove print statement
1 parent 7fa0ad2 commit 432557a

File tree

2 files changed

+1
-2
lines changed

2 files changed

+1
-2
lines changed

audio_diffusion_pytorch/model.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -120,7 +120,6 @@ def forward( # type: ignore
120120
self, x: Tensor, with_info: bool = False, **kwargs
121121
) -> Union[Tensor, Tuple[Tensor, Any]]:
122122
latent, info = self.encode(x, with_info=True)
123-
print(latent.shape)
124123
loss = super().forward(x, channels_list=[latent], **kwargs)
125124
return (loss, info) if with_info else loss
126125

setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
setup(
44
name="audio-diffusion-pytorch",
55
packages=find_packages(exclude=[]),
6-
version="0.0.96",
6+
version="0.0.97",
77
license="MIT",
88
description="Audio Diffusion - PyTorch",
99
long_description_content_type="text/markdown",

0 commit comments

Comments
 (0)