fix: patch factor missing from autoencoder, set default

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

Hide file tree

Showing 4 changed files with 6 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -126,6 +126,7 @@ unet = UNet1d(
     in_channels=1,
     channels=128,
     patch_blocks=4,
+    patch_factor=2,
     kernel_sizes_init=[1, 3, 7],
     multipliers=[1, 2, 4, 4, 4, 4, 4],
     factors=[4, 4, 4, 2, 2, 2],

diff --git a/audio_diffusion_pytorch/model.py b/audio_diffusion_pytorch/model.py
@@ -109,6 +109,7 @@ def __init__(
         in_channels: int,
         channels: int,
         patch_blocks: int,
+        patch_factor: int,
         kernel_sizes_init: Sequence[int],
         multipliers: Sequence[int],
         factors: Sequence[int],
@@ -130,6 +131,7 @@ def __init__(
             in_channels=in_channels,
             channels=channels,
             patch_blocks=patch_blocks,
+            patch_factor=patch_factor,
             num_layers=encoder_depth,
             latent_channels=encoder_channels,
             multipliers=multipliers,
@@ -144,6 +146,7 @@ def __init__(
             in_channels=in_channels,
             channels=channels,
             patch_blocks=patch_blocks,
+            patch_factor=patch_factor,
             kernel_sizes_init=kernel_sizes_init,
             multipliers=multipliers,
             factors=factors,

diff --git a/audio_diffusion_pytorch/modules.py b/audio_diffusion_pytorch/modules.py
@@ -875,6 +875,7 @@ def __init__(
         in_channels: int,
         channels: int,
         patch_blocks: int,
+        patch_factor: int,
         multipliers: Sequence[int],
         factors: Sequence[int],
         num_blocks: Sequence[int],
@@ -890,7 +891,6 @@ def __init__(
         use_attention_bottleneck: bool,
         use_context_time: bool,
         out_channels: Optional[int] = None,
-        patch_factor: int = 2,
         context_features: Optional[int] = None,
         context_channels: Optional[Sequence[int]] = None,
         context_embedding_features: Optional[int] = None,

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