lt-tensor 0.0.1.dev1__py3-none-any.whl → 0.0.1.dev3__py3-none-any.whl

lt_tensor/__init__.py

@@ -1 +1 @@
-__version__ = "0.0.1dev1"
+__version__ = "0.0.1dev3"

lt_tensor/misc_utils.py

@@ -23,7 +23,7 @@ def log_tensor(
         _b = 20 + len(title.strip())
     print(f"shape: {item.shape}")
     print(f"dtype: {item.dtype}")
-    if not print_details:
+    if print_details:
         print(f"ndim: {item.ndim}")
         if isinstance(item, Tensor):
             print(f"device: {item.device}")

lt_tensor/transform.py

@@ -175,7 +175,7 @@ def get_sinusoidal_embedding(timesteps: torch.Tensor, dim: int) -> torch.Tensor:
     return emb
 
 
-def generate_window(
+def _generate_window(
     M: int, alpha: float = 0.5, device: Optional[DeviceType] = None
 ) -> Tensor:
     if M < 1:
@@ -260,7 +260,7 @@ def window_sumsquare(
     x = torch.zeros(total_length, dtype=dtype, device=device)
 
     # Get the window (from scipy for now)
-    win = generate_window(window_spec, win_length, fftbins=True)
+    win = _generate_window(window_spec, win_length, fftbins=True)
     win = torch.tensor(win, dtype=dtype, device=device)
 
     # Normalize and square
@@ -277,9 +277,6 @@ def window_sumsquare(
     return x
 
 
-def get_window(win_length: int = 1200):
-    return generate_window(win_length)
-
 
 def inverse_transform(
     spec: Tensor,
@@ -288,10 +285,10 @@ def inverse_transform(
     n_fft: int = 2048,
     hop_length: int = 300,
     win_length: int = 1200,
-    length: Optional[torch.shape] = None,
+    length: Optional[Any] = None,
 ):
     if window is None:
-        window = generate_window(win_length)
+        window = _generate_window(win_length)
     return torch.istft(
         spec * torch.exp(phase * 1j),
         n_fft,
@@ -313,7 +310,7 @@ def stft_istft_rebuild(
     Perform STFT followed by ISTFT reconstruction using magnitude and phase.
     """
     if window is None:
-        window = generate_window(win_length)
+        window = _generate_window(win_length)
     st = torch.stft(
         input_data,
         n_fft,

{lt_tensor-0.0.1.dev1.dist-info → lt_tensor-0.0.1.dev3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lt-tensor
-Version: 0.0.1.dev1
+Version: 0.0.1.dev3
 Summary: General utilities for PyTorch and others. Built for general use.
 Home-page: https://github.com/gr1336/lt-tensor/
 Author: gr1336

{lt_tensor-0.0.1.dev1.dist-info → lt_tensor-0.0.1.dev3.dist-info}/RECORD

@@ -1,11 +1,11 @@
-lt_tensor/__init__.py,sha256=fHS3r6eTARkZmS9vI1iPgHEL-H8y3z0skMRkAfilD8g,26
+lt_tensor/__init__.py,sha256=pUB05ZkgkpP10ivzwoWdbq_HCxw-iOsbf6m8eFtx-YM,26
 lt_tensor/_basics.py,sha256=XS2OrvyzboUTKURGMU1fmbn2gFq2779HW8xz05fm4x8,8181
 lt_tensor/_torch_commons.py,sha256=_2Eck-MsQ46PxW5ku7NJvNSL5vg54_4GkLCqdzFevwA,402
 lt_tensor/lr_schedulers.py,sha256=oLYw2X78KSdlOD0pwPO0lsBj1xqLOAzT8rnRBnCO19o,3560
 lt_tensor/math_ops.py,sha256=4jUTtS1ZwJnh1AaFGhMuKTcakjwLQ846AKcBoDEwudI,1862
-lt_tensor/misc_utils.py,sha256=EaQ__986n5J-oAnMXmTp9hXxCDk6NrgvGlU_C05M7G4,19193
+lt_tensor/misc_utils.py,sha256=Zo9p-iwRLNwVBk5kRP5sPCw7IiOBST9hW7LBK9JKCVs,19189
 lt_tensor/monotonic_align.py,sha256=LhBd8p1xdBzg6jQrQX1j7b4PNeYGwIqM24zcU-pHOLE,2239
-lt_tensor/transform.py,sha256=J6KFmOmQCeCpbK7r8yaUZ8RiW-o-rvtSFvBVM5utBhk,9208
+lt_tensor/transform.py,sha256=CbTzqRUzl-U-2KMHtYUgiliU_61H3cag8a-N526nzdg,9124
 lt_tensor/model_zoo/__init__.py,sha256=sIidI3gSoxTq2OTenqADiFlmd9EH8rS_mzjLePdNHqc,77
 lt_tensor/model_zoo/basic.py,sha256=Ccjg26-gnqbkOUg0aNnLhZmsfvMwqTxXVXimfC5qnl0,1862
 lt_tensor/model_zoo/residual.py,sha256=iViJ5MrIY1vUbPB0ZiOjh62wfQagbPulMvbZPTXp-OU,7370
@@ -14,8 +14,8 @@ lt_tensor/model_zoo/diffusion/models.py,sha256=LRf5B2MPic4Dwfvg2PxG61KFtqvATV3wO
 lt_tensor/model_zoo/transformer_models/__init__.py,sha256=NOvCP-EySkNHLbGOxFMURp0AztvJQSNo9J_0hsEzLto,130
 lt_tensor/model_zoo/transformer_models/models.py,sha256=Xh2nq83w0qMX-Co7EmaQb_auq7Fi7kyULklc0_yq3oo,4088
 lt_tensor/model_zoo/transformer_models/positional_encoders.py,sha256=ilJDKAonnU0973BT_7gT5ke_8PbWHMxsr8EvNQHAcUY,3529
-lt_tensor-0.0.1.dev1.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-lt_tensor-0.0.1.dev1.dist-info/METADATA,sha256=Os0vIySmcMifWqS32PHnLRTz2dW_VfF-tf_jhNcmPBk,1055
-lt_tensor-0.0.1.dev1.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
-lt_tensor-0.0.1.dev1.dist-info/top_level.txt,sha256=35FuhFeXnUyvHWdbVHGPh0hS8euofafnJ_GJAVSF4Kk,10
-lt_tensor-0.0.1.dev1.dist-info/RECORD,,
+lt_tensor-0.0.1.dev3.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+lt_tensor-0.0.1.dev3.dist-info/METADATA,sha256=cIBisih0DD2ZXdcHxjEUofVmPG8L0BxhIiym7AIkVSo,1055
+lt_tensor-0.0.1.dev3.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
+lt_tensor-0.0.1.dev3.dist-info/top_level.txt,sha256=35FuhFeXnUyvHWdbVHGPh0hS8euofafnJ_GJAVSF4Kk,10
+lt_tensor-0.0.1.dev3.dist-info/RECORD,,