lt-tensor 0.0.1a30__tar.gz → 0.0.1a32__tar.gz

{lt_tensor-0.0.1a30 → lt_tensor-0.0.1a32}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lt-tensor
-Version: 0.0.1a30
+Version: 0.0.1a32
 Summary: General utilities for PyTorch and others. Built for general use.
 Home-page: https://github.com/gr1336/lt-tensor/
 Author: gr1336
@@ -17,7 +17,7 @@ Requires-Dist: numpy>=1.26.4
 Requires-Dist: tokenizers
 Requires-Dist: pyyaml>=6.0.0
 Requires-Dist: numba>0.60.0
-Requires-Dist: lt-utils==0.0.2
+Requires-Dist: lt-utils>=0.0.3
 Requires-Dist: librosa==0.11.*
 Requires-Dist: einops
 Requires-Dist: plotly

{lt_tensor-0.0.1a30 → lt_tensor-0.0.1a32}/lt_tensor/config_templates.py

@@ -18,11 +18,11 @@ class ModelConfig(ABC, OrderedDict):
 
     def reset_settings(self):
         raise NotImplementedError("Not implemented")
-    
+
     def post_process(self):
         """Implement the post process, to do a final check to the input data"""
         pass
-    
+
     def save_config(
         self,
         path: str,
@@ -58,12 +58,12 @@ class ModelConfig(ABC, OrderedDict):
     def from_dict(
         cls,
         dictionary: Dict[str, Any],
-    ) -> "ModelConfig":
+    ):
         assert is_dict(dictionary)
-        return ModelConfig(**dictionary)
+        return cls(**dictionary)
 
     @classmethod
-    def from_path(cls, path_name: PathLike) -> "ModelConfig":
+    def from_path(cls, path_name: PathLike):
         assert is_file(path_name) or is_dir(path_name)
         settings = {}
 
@@ -90,6 +90,4 @@ class ModelConfig(ABC, OrderedDict):
         settings.pop("path", None)
         settings.pop("path_name", None)
 
-        return ModelConfig(**settings)
-
-
+        return cls(**settings)

{lt_tensor-0.0.1a30 → lt_tensor-0.0.1a32}/lt_tensor/model_zoo/losses/discriminators.py

@@ -20,6 +20,7 @@ def get_padding(kernel_size, dilation=1):
 
 class MultiDiscriminatorWrapper(ConvNets):
     """Base for all multi-steps type of discriminators"""
+
     def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)
         self.leaky_relu = nn.LeakyReLU(kwargs.get("negative_slope", 0.1))
@@ -207,7 +208,7 @@ class MultiPeriodDiscriminator(MultiDiscriminatorWrapper):
         return y_d_rs, y_d_gs, fmap_rs, fmap_gs
 
 
-class EnvelopeExtractor(nn.Module):
+class EnvelopeExtractor(Model):
     """Extracts the amplitude envelope of the audio signal."""
 
     def __init__(self, kernel_size=101):
@@ -216,7 +217,7 @@ class EnvelopeExtractor(nn.Module):
         self.kernel_size = kernel_size
         self.register_buffer("kernel", torch.ones(1, 1, kernel_size) / kernel_size)
 
-    def forward(self, x):
+    def forward(self, x: Tensor):
         # x: (B, 1, T) -> abs(x)
         envelope = torch.abs(x)
         # Apply low-pass smoothing (via conv1d)
@@ -274,7 +275,6 @@ class MultiEnvelopeDiscriminator(MultiDiscriminatorWrapper):
     def forward(self, y, y_hat):
         y_d_rs, y_d_gs = [], []
         fmap_rs, fmap_gs = [], []
-
         for i, d in enumerate(self.discriminators):
             if i != 0:
                 y = self.meanpools[i - 1](y)
@@ -555,7 +555,19 @@ class MultiResolutionDiscriminator(MultiDiscriminatorWrapper):
 
 
 class MultiDiscriminatorStep(Model):
-    def __init__(self, list_discriminator: List[MultiDiscriminatorWrapper]):
+    def __init__(
+        self, list_discriminator: List[MultiDiscriminatorWrapper]
+    ):
+        """Setup example:
+        model_d = MultiDiscriminatorStep(
+            [
+                MultiEnvelopeDiscriminator(),
+                MultiBandDiscriminator(),
+                MultiResolutionDiscriminator(),
+                MultiPeriodDiscriminator(0.5),
+            ]
+        )
+        """
         super().__init__()
         self.disc: Sequence[MultiDiscriminatorWrapper] = nn.ModuleList(
             list_discriminator

{lt_tensor-0.0.1a30 → lt_tensor-0.0.1a32}/lt_tensor.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lt-tensor
-Version: 0.0.1a30
+Version: 0.0.1a32
 Summary: General utilities for PyTorch and others. Built for general use.
 Home-page: https://github.com/gr1336/lt-tensor/
 Author: gr1336
@@ -17,7 +17,7 @@ Requires-Dist: numpy>=1.26.4
 Requires-Dist: tokenizers
 Requires-Dist: pyyaml>=6.0.0
 Requires-Dist: numba>0.60.0
-Requires-Dist: lt-utils==0.0.2
+Requires-Dist: lt-utils>=0.0.3
 Requires-Dist: librosa==0.11.*
 Requires-Dist: einops
 Requires-Dist: plotly

{lt_tensor-0.0.1a30 → lt_tensor-0.0.1a32}/lt_tensor.egg-info/requires.txt

@@ -4,7 +4,7 @@ numpy>=1.26.4
 tokenizers
 pyyaml>=6.0.0
 numba>0.60.0
-lt-utils==0.0.2
+lt-utils>=0.0.3
 librosa==0.11.*
 einops
 plotly

{lt_tensor-0.0.1a30 → lt_tensor-0.0.1a32}/setup.py

@@ -4,7 +4,7 @@ with open("README.md", "r", encoding="utf-8") as f:
     long_description = f.read()
 
 setup(
-    version="0.0.1a30",
+    version="0.0.1a32",
     name="lt-tensor",
     description="General utilities for PyTorch and others. Built for general use.",
     long_description=long_description,
@@ -17,7 +17,7 @@ setup(
         "tokenizers",
         "pyyaml>=6.0.0",
         "numba>0.60.0",
-        "lt-utils==0.0.2",
+        "lt-utils>=0.0.3",
         "librosa==0.11.*",
         "einops",
         "plotly",