feat: add option to change context_features_multiplier

archinetai · Nov 16, 2022 · 5fb0939 · 5fb0939
1 parent fdf55ba
commit 5fb0939
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/audio_diffusion_pytorch/modules.py b/audio_diffusion_pytorch/modules.py
@@ -938,6 +938,7 @@ def __init__(
         use_stft_context: bool = False,
         out_channels: Optional[int] = None,
         context_features: Optional[int] = None,
+        context_features_multiplier: int = 4,
         context_channels: Optional[Sequence[int]] = None,
         context_embedding_features: Optional[int] = None,
         **kwargs,
@@ -975,7 +976,7 @@ def __init__(
         )
 
         if use_context_time or use_context_features:
-            context_mapping_features = channels * 4
+            context_mapping_features = channels * context_features_multiplier
 
             self.to_mapping = nn.Sequential(
                 nn.Linear(context_mapping_features, context_mapping_features),

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