fix a small bug thanks to @Lunariz

cmu-mlsp · Feb 9, 2023 · c8fe964 · c8fe964
1 parent 6ed3aad
commit c8fe964
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/audiolm_pytorch/soundstream.py b/audiolm_pytorch/soundstream.py
@@ -145,6 +145,7 @@ def __init__(
 
     def forward(self, x):
         weight, bias = map(torch.view_as_complex, (self.weight, self.bias))
+        print(x.dtype, weight.dtype, x.device, weight.device)
         return F.conv2d(x, weight, bias, stride = self.stride, padding = self.padding)
 
 def ComplexSTFTResidualUnit(chan_in, chan_out, strides):

diff --git a/audiolm_pytorch/trainer.py b/audiolm_pytorch/trainer.py
@@ -951,8 +951,7 @@ def __init__(
                 folder,
                 max_length = data_max_length,
                 target_sample_hz = soundstream.target_sample_hz,
-                seq_len_multiple_of = soundstream.seq_len_multiple_of,
-                normalize = dataset_normalize
+                seq_len_multiple_of = soundstream.seq_len_multiple_of
             )
 
         self.ds_fields = None

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