fix: missing patch factor in patchblock

archinetai · Sep 23, 2022 · a6df6bd · a6df6bd
1 parent c76e8a6
commit a6df6bd
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/audio_diffusion_pytorch/modules.py b/audio_diffusion_pytorch/modules.py
@@ -303,6 +303,7 @@ def __init__(
             PatchBlock(
                 in_channels=channels_pre[i],
                 out_channels=channels_post[i],
+                patch_size=factor,
                 context_mapping_features=context_mapping_features,
             )
             for i in range(blocks)
@@ -328,6 +329,7 @@ def __init__(
             UnpatchBlock(
                 in_channels=channels_pre[i],
                 out_channels=channels_post[i],
+                patch_size=factor,
                 context_mapping_features=context_mapping_features,
             )
             for i in range(blocks)

diff --git a/setup.py b/setup.py
@@ -3,7 +3,7 @@
 setup(
     name="audio-diffusion-pytorch",
     packages=find_packages(exclude=[]),
-    version="0.0.41",
+    version="0.0.42",
     license="MIT",
     description="Audio Diffusion - PyTorch",
     long_description_content_type="text/markdown",