fix: remove unused kernel_sizes_init

Author: flavioschneider

README.md

@@ -127,7 +127,6 @@ unet = UNet1d(
     channels=128,
     patch_blocks=16,
     patch_factor=1,
-    kernel_sizes_init=[1, 3, 7],
     multipliers=[1, 2, 4, 4, 4, 4, 4],
     factors=[4, 4, 4, 2, 2, 2],
     attentions=[0, 0, 0, 1, 1, 1, 1],

audio_diffusion_pytorch/model.py

@@ -110,7 +110,6 @@ def __init__(
         channels: int,
         patch_blocks: int,
         patch_factor: int,
-        kernel_sizes_init: Sequence[int],
         multipliers: Sequence[int],
         factors: Sequence[int],
         num_blocks: Sequence[int],
@@ -137,7 +136,6 @@ def __init__(
             multipliers=multipliers,
             factors=factors,
             num_blocks=encoder_num_blocks,
-            kernel_sizes_init=kernel_sizes_init,
             kernel_multiplier_downsample=kernel_multiplier_downsample,
             resnet_groups=resnet_groups,
         )
@@ -147,7 +145,6 @@ def __init__(
             channels=channels,
             patch_blocks=patch_blocks,
             patch_factor=patch_factor,
-            kernel_sizes_init=kernel_sizes_init,
             multipliers=multipliers,
             factors=factors,
             num_blocks=num_blocks,
@@ -204,7 +201,6 @@ def get_default_model_kwargs():
         channels=128,
         patch_blocks=1,
         patch_factor=16,
-        kernel_sizes_init=[1, 3, 7],
         multipliers=[1, 2, 4, 4, 4, 4, 4],
         factors=[4, 4, 4, 2, 2, 2],
         num_blocks=[2, 2, 2, 2, 2, 2],

audio_diffusion_pytorch/modules.py

@@ -806,7 +806,6 @@ def __init__(
         attention_multiplier: int,
         resnet_groups: int,
         kernel_multiplier_downsample: int,
-        kernel_sizes_init: Sequence[int],
         use_nearest_upsample: bool,
         use_skip_scale: bool,
         use_context_time: bool,
@@ -1109,7 +1108,6 @@ def __init__(
         patch_blocks: int,
         resnet_groups: int,
         kernel_multiplier_downsample: int,
-        kernel_sizes_init: Sequence[int],
         num_layers: int,
         latent_channels: int,
         multipliers: Sequence[int],

setup.py

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