feat: fix upsampler kwargs

Author: flavioschneider

README.md

@@ -174,6 +174,7 @@ y_long = composer(y, keep_start=True) # [1, 1, 98304]
 - [x] Add (variational) autoencoder option to compress audio before diffusion.
 - [x] Fix inpainting and make it work with ADPM2 sampler.
 - [x] Add trainer with experiments.
+- [x] Add diffusion upsampler.
 
 ## Appreciation
 

audio_diffusion_pytorch/model.py

@@ -152,7 +152,7 @@ def __init__(self, factor: int, in_channels: int = 1, *args, **kwargs):
             context_channels=[in_channels],
         )
 
-        super().__init__(*args, {**default_kwargs, **kwargs})  # type: ignore
+        super().__init__(*args, **{**default_kwargs, **kwargs})  # type: ignore
 
     def forward(self, x: Tensor, **kwargs) -> Tensor:
         # Downsample by picking every `factor` item

setup.py

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