feat: fewer channels in convout block

Author: flavioschneider

audio_diffusion_pytorch/modules.py

@@ -222,7 +222,7 @@ def __init__(
         context_mapping_features: Optional[int] = None,
     ):
         super().__init__()
-        mid_channels = in_channels * 32
+        mid_channels = in_channels * 8
 
         self.layers = nn.ModuleList(
             ResnetBlock1d(

setup.py

@@ -3,7 +3,7 @@
 setup(
     name="audio-diffusion-pytorch",
     packages=find_packages(exclude=[]),
-    version="0.0.37",
+    version="0.0.38",
     license="MIT",
     description="Audio Diffusion - PyTorch",
     long_description_content_type="text/markdown",