fix a bug thanks to @eonglints

lucidrains · Nov 15, 2022 · 2725ae8 · 2725ae8
1 parent 5f96496
commit 2725ae8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/audiolm_pytorch/audiolm_pytorch.py b/audiolm_pytorch/audiolm_pytorch.py
@@ -109,7 +109,7 @@ def __init__(
             chan_out = min(curr_channels * 4, chan_max)
 
             self.conv_layers.append(nn.Sequential(
-                nn.Conv1d(curr_channels, chan_out, 8, stride = 4, padding = 4),
+                nn.Conv1d(curr_channels, chan_out, 8, stride = 4, padding = 4, groups = groups),
                 leaky_relu()
             ))
 

diff --git a/setup.py b/setup.py
@@ -3,7 +3,7 @@
 setup(
   name = 'audiolm-pytorch',
   packages = find_packages(exclude=[]),
-  version = '0.0.19',
+  version = '0.0.20',
   license='MIT',
   description = 'AudioLM - Language Modeling Approach to Audio Generation from Google Research - Pytorch',
   author = 'Phil Wang',