fix: module list for bottlenecks

Author: flavioschneider

audio_diffusion_pytorch/modules.py

@@ -1381,7 +1381,7 @@ def __init__(
     ):
         super().__init__()
         num_layers = len(multipliers) - 1
-        self.bottlenecks = to_list(bottleneck)
+        self.bottlenecks = nn.ModuleList(to_list(bottleneck))
         self.use_noisy = use_noisy
         self.use_magnitude_channels = use_magnitude_channels
 

setup.py

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