flamo 0.1.12__py3-none-any.whl → 0.2.0__py3-none-any.whl

flamo/processor/system.py

@@ -26,7 +26,7 @@ class Series(nn.Sequential):
         # Check nfft and alpha values
         self.nfft = self.__check_attribute("nfft")
         self.alias_decay_db = self.__check_attribute("alias_decay_db")
-
+        self.dtype = self.__check_attribute("dtype")
         # Check I/O compatibility
         self.input_channels, self.output_channels = self.__check_io()
 
@@ -65,6 +65,7 @@ class Series(nn.Sequential):
         # Check nfft and alpha values
         self.nfft = self.__check_attribute("nfft")
         self.alias_decay_db = self.__check_attribute("alias_decay_db")
+        self.dtype = self.__check_attribute("dtype")
 
         # Check I/O compatibility
         self.input_channels, self.output_channels = self.__check_io()
@@ -116,6 +117,7 @@ class Series(nn.Sequential):
         # Check nfft and alpha values
         self.nfft = self.__check_attribute("nfft")
         self.alias_decay_db = self.__check_attribute("alias_decay_db")
+        self.dtype = self.__check_attribute("dtype")
 
         # Check I/O compatibility
         self.input_channels, self.output_channels = self.__check_io()
@@ -357,7 +359,7 @@ class Recursion(nn.Module):
         # Check nfft and time anti-aliasing decay-envelope parameter values
         self.nfft = self.__check_attribute("nfft")
         self.alias_decay_db = self.__check_attribute("alias_decay_db")
-
+        self.dtype = self.__check_attribute("dtype")
         # Check I/O compatibility
         self.input_channels, self.output_channels = self.__check_io()
 
@@ -402,7 +404,7 @@ class Recursion(nn.Module):
                 torch.Tensor: The identity matrix.
         """
         size = (self.nfft // 2 + 1, self.output_channels, self.output_channels)
-        I = torch.complex(torch.zeros(*size), torch.zeros(*size))
+        I = torch.complex(torch.zeros(*size, dtype=self.dtype), torch.zeros(*size, dtype=self.dtype))
         for i in range(self.output_channels):
             I[:, i, i] = 1
         return I
@@ -544,6 +546,7 @@ class Parallel(nn.Module):
         # Check nfft and time anti-aliasing decay-envelope parameter values
         self.nfft = self.__check_attribute("nfft")
         self.alias_decay_db = self.__check_attribute("alias_decay_db")
+        self.dtype = self.__check_attribute("dtype")
 
         # Check I/O compatibility
         self.input_channels, self.output_channels = self.__check_io()
@@ -719,7 +722,7 @@ class Shell(nn.Module):
         # Check model nfft and time anti-aliasing decay-envelope parameter values
         self.nfft = self.__check_attribute("nfft")
         self.alias_decay_db = self.__check_attribute("alias_decay_db")
-
+        self.dtype = self.__check_attribute("dtype")
         # Check I/O compatibility
         self.input_channels, self.output_channels = self.__check_io()
 
@@ -895,15 +898,15 @@ class Shell(nn.Module):
             .view(1, -1, 1)
             .expand(1, -1, self.output_channels)
         )
-
+        self.alias_envelope = self.alias_envelope.to(dtype=self.dtype)
         # save input/output layers
         input_save = self.get_inputLayer()
         output_save = self.get_outputLayer()
 
         # update input/output layers
-        self.set_inputLayer(FFT(self.nfft))
+        self.set_inputLayer(FFT(self.nfft, dtype=self.dtype))
         self.set_outputLayer(
-            nn.Sequential(iFFT(self.nfft), Transform(lambda x: x * self.alias_envelope))
+            nn.Sequential(iFFT(self.nfft, dtype=self.dtype), Transform(lambda x: x * self.alias_envelope))
         )
 
         # generate input signal
@@ -914,6 +917,7 @@ class Shell(nn.Module):
             signal_type="impulse",
             fs=fs,
             device=gamma.device,
+            dtype=self.dtype
         )
         if identity and self.input_channels > 1:
             self.alias_envelope = self.alias_envelope.unsqueeze(-1).expand(
@@ -963,7 +967,7 @@ class Shell(nn.Module):
             .view(1, -1, 1)
             .expand(1, -1, self.output_channels)
         )
-
+        self.alias_envelope_exp = self.alias_envelope_exp.to(dtype=self.dtype)
         # save input/output layers
         input_save = self.get_inputLayer()
         output_save = self.get_outputLayer()
@@ -972,13 +976,13 @@ class Shell(nn.Module):
         self.set_inputLayer(FFT(self.nfft))
         self.set_outputLayer(
             nn.Sequential(
-                iFFT(self.nfft),
+                iFFT(self.nfft, dtype=self.dtype),
                 Transform(
                     lambda x: torch.einsum(
                         "bfm..., bfm... -> bfm...", x, self.alias_envelope_exp
                     )
                 ),
-                FFT(self.nfft),
+                FFT(self.nfft, dtype=self.dtype),
             )
         )  # TODO, this is a very suboptimal way to do this, we need to find a better way
 
@@ -990,6 +994,7 @@ class Shell(nn.Module):
             signal_type="impulse",
             fs=fs,
             device=gamma.device,
+            dtype=self.dtype
         )
         if identity and self.input_channels > 1:
             x = x.diag_embed()

flamo/utils.py

@@ -22,12 +22,12 @@ def to_complex(x):
     return torch.complex(x, torch.zeros_like(x))
 
 
-def save_audio(filepath, x, fs=48000):
+def save_audio(filepath, x, fs=48000, subtype='PCM_24'):
     # check if the folder exists
     folder = os.path.dirname(filepath)
     if not os.path.exists(folder):
         os.makedirs(folder)
-    sf.write(filepath, x.cpu().numpy(), fs)
+    sf.write(filepath, x.cpu().numpy(), fs, subtype=subtype)
 
 
 class RegularGridInterpolator:

{flamo-0.1.12.dist-info → flamo-0.2.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: flamo
-Version: 0.1.12
+Version: 0.2.0
 Summary: An Open-Source Library for Frequency-Domain Differentiable Audio Processing
 Project-URL: Homepage, https://github.com/gdalsanto/flamo
 Project-URL: Issues, https://github.com/gdalsanto/flamo/issues

flamo-0.2.0.dist-info/RECORD

@@ -0,0 +1,24 @@
+flamo/__init__.py,sha256=ujezWOJfD7DUoj4q1meeMUnB97rOEtNR7mYw_PE9LMg,49
+flamo/functional.py,sha256=3vO5IHAO0kD0icNZw-EX5pecsmSUza-GmuY3WjHxN1M,38846
+flamo/utils.py,sha256=Gn0aVa32lKeDcN0buyDGErq_uT7dUBmkEXweXpz-3Zw,3385
+flamo/auxiliary/__init__.py,sha256=7lVNh8OxHloZ4KPmp-iTUJnUbi8XbuRzGaQ3Z-NKXio,42
+flamo/auxiliary/eq.py,sha256=aI0og6x82r_o3Q7xGu36scdf51e2ARhYmFyJqnEIzQc,7364
+flamo/auxiliary/filterbank.py,sha256=02w8dI8HoNDtKpdVhSJkIkd-h-KNXvZtivf3l4_ozzU,9866
+flamo/auxiliary/minimize.py,sha256=fMTAAAk9yD7Y4luKS4XA1-HTq44xo2opq_dRPRrhlIY,2474
+flamo/auxiliary/reverb.py,sha256=jrNsNUH5B0S-1U-DfPnNVGqpbKCZHfeiFmoCGeiX1QU,34915
+flamo/auxiliary/scattering.py,sha256=IV9ZgkMfgEWtQo3IeLJviAXAdjrYI_Tob-H_8Zv8oBA,9729
+flamo/auxiliary/velvet.py,sha256=B4pYEnhaQPkh02pxqiGdAhLRX2g-eWtHezphi0_h4Qs,4201
+flamo/auxiliary/config/config.py,sha256=sZ3XvqwV6KiIc2n8HRtg7YJE3zhc7Vqblbqs-Z0bsKg,2978
+flamo/optimize/__init__.py,sha256=grgxLmQ7m-c9MvRdIejmEAaaajfBwgeaZAv2qjHIvPw,65
+flamo/optimize/dataset.py,sha256=cdedwSZRASIfk21zSlflerpKF1TmVy0JWDw8dhSA4z4,5976
+flamo/optimize/loss.py,sha256=h6EeqjdX5P1SqDBKBavSxV25VBgnYK8tuX91wk6lw_g,33466
+flamo/optimize/surface.py,sha256=aYWXg2ZNJzFjHRILcgrTfNkQ2VSoaMfMSnu9IPM8O-8,26475
+flamo/optimize/trainer.py,sha256=he4nUjLC-3RTlxxBIw33r5k8mQfgAGvN1wpPBAWCjVo,12045
+flamo/optimize/utils.py,sha256=R5-KoZagRho3eykY88pC3UB2mc5SsE4Yv9X-ogskXdA,1610
+flamo/processor/__init__.py,sha256=paGdxGVZgA2VAs0tBwRd0bobzGxeyK79DS7ZGO8drkI,41
+flamo/processor/dsp.py,sha256=a5etio4qfJWwAsHpAk3CXDgEd3LMet-r_956216SO6M,128874
+flamo/processor/system.py,sha256=Hct-o6IgF5NQ2xYbX-1j3st94hMoM8dOgAzle2gjDqU,43145
+flamo-0.2.0.dist-info/METADATA,sha256=cGqrN7snqzo-GWh2jLQyEXqQHCuPa0BNPRW5Y0IgwTA,7825
+flamo-0.2.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+flamo-0.2.0.dist-info/licenses/LICENSE,sha256=smMocRH7xdPT5RvFNqSLtbSNzohXJM5G_rX1Qaej6vg,1120
+flamo-0.2.0.dist-info/RECORD,,

flamo-0.1.12.dist-info/RECORD

@@ -1,24 +0,0 @@
-flamo/__init__.py,sha256=ujezWOJfD7DUoj4q1meeMUnB97rOEtNR7mYw_PE9LMg,49
-flamo/functional.py,sha256=8cKYzR5GDzH4ph4ja7x1znNFL-8ufWoxa666QHaZR1A,36917
-flamo/utils.py,sha256=ypGKSABZMphgIrjCKgCH-zgR7BaupRbyzuUhsZFqAAM,3350
-flamo/auxiliary/__init__.py,sha256=7lVNh8OxHloZ4KPmp-iTUJnUbi8XbuRzGaQ3Z-NKXio,42
-flamo/auxiliary/eq.py,sha256=eIWMIq0ggizXLhTdeWWbgBXWUFXCJyoEbkBH7Gzasao,6779
-flamo/auxiliary/filterbank.py,sha256=02w8dI8HoNDtKpdVhSJkIkd-h-KNXvZtivf3l4_ozzU,9866
-flamo/auxiliary/minimize.py,sha256=fMTAAAk9yD7Y4luKS4XA1-HTq44xo2opq_dRPRrhlIY,2474
-flamo/auxiliary/reverb.py,sha256=fAJgTGq_yaQguBrrnifJNl4ySIkPb0NdrcFKNCNL4xc,34769
-flamo/auxiliary/scattering.py,sha256=qlK8cynrpde56yLlbPuScC0Y1VmsPb0SFXl6Xisv6hA,9420
-flamo/auxiliary/velvet.py,sha256=B4pYEnhaQPkh02pxqiGdAhLRX2g-eWtHezphi0_h4Qs,4201
-flamo/auxiliary/config/config.py,sha256=CxXj-8sLq0_m9KyLg1a6NwLoK1UvTz3i0jZOLraq14I,2893
-flamo/optimize/__init__.py,sha256=grgxLmQ7m-c9MvRdIejmEAaaajfBwgeaZAv2qjHIvPw,65
-flamo/optimize/dataset.py,sha256=2mfzsnyX_bzavXouII9ee_pd6ti4lv215ieGJHscceI,5829
-flamo/optimize/loss.py,sha256=h6EeqjdX5P1SqDBKBavSxV25VBgnYK8tuX91wk6lw_g,33466
-flamo/optimize/surface.py,sha256=eGDg9Zksicu1EO3irjyECEhp2oplj9HMNNuDK_WnHe4,26139
-flamo/optimize/trainer.py,sha256=he4nUjLC-3RTlxxBIw33r5k8mQfgAGvN1wpPBAWCjVo,12045
-flamo/optimize/utils.py,sha256=R5-KoZagRho3eykY88pC3UB2mc5SsE4Yv9X-ogskXdA,1610
-flamo/processor/__init__.py,sha256=paGdxGVZgA2VAs0tBwRd0bobzGxeyK79DS7ZGO8drkI,41
-flamo/processor/dsp.py,sha256=n92YJPrES-ydwHgXmZ9RkFevIC3n-Wh4X8I1QNZqcV0,126378
-flamo/processor/system.py,sha256=9XwLtaGEVs9glVOFvyiPnQpsnR_Wjrv6k1i1qCs8D1Q,42516
-flamo-0.1.12.dist-info/METADATA,sha256=03qRhouKUzsmZs4bXb3LUGyhVAUIYLiZk6Z7KOiYyyk,7826
-flamo-0.1.12.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-flamo-0.1.12.dist-info/licenses/LICENSE,sha256=smMocRH7xdPT5RvFNqSLtbSNzohXJM5G_rX1Qaej6vg,1120
-flamo-0.1.12.dist-info/RECORD,,