lt-tensor 0.0.1a20__tar.gz → 0.0.1a21__tar.gz

{lt_tensor-0.0.1a20 → lt_tensor-0.0.1a21}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lt-tensor
-Version: 0.0.1a20
+Version: 0.0.1a21
 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.2a3
+Requires-Dist: lt-utils==0.0.2
 Requires-Dist: librosa==0.11.*
 Requires-Dist: einops
 Requires-Dist: plotly

{lt_tensor-0.0.1a20 → lt_tensor-0.0.1a21}/lt_tensor/model_base.py

@@ -207,12 +207,12 @@ class Model(_Devices_Base, ABC):
     def autocast(self, value: bool):
         self._autocast = value
 
-    def freeze_all(self, exclude: Optional[List[str]] = None):
+    def freeze_all(self, exclude: Optional[List[str]] = None, force: bool = False):
         no_exclusions = not exclude
         no_exclusions = not exclude
         results = []
         for name, module in self.named_modules():
-            if name not in self.registered_freezable_modules:
+            if not force and name not in self.registered_freezable_modules:
                 results.append(
                     (
                         name,
@@ -228,12 +228,13 @@ class Model(_Devices_Base, ABC):
                 results.append((name, "excluded"))
         return results
 
-    def unfreeze_all(self, exclude: Optional[list[str]] = None):
+    def unfreeze_all(self, exclude: Optional[list[str]] = None, force: bool = False):
         """Unfreezes all model parameters except specified layers."""
         no_exclusions = not exclude
         results = []
         for name, module in self.named_modules():
-            if name not in self.registered_freezable_modules:
+            if not force and name not in self.registered_freezable_modules:
+
                 results.append(
                     (
                         name,

{lt_tensor-0.0.1a20 → lt_tensor-0.0.1a21}/lt_tensor.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lt-tensor
-Version: 0.0.1a20
+Version: 0.0.1a21
 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.2a3
+Requires-Dist: lt-utils==0.0.2
 Requires-Dist: librosa==0.11.*
 Requires-Dist: einops
 Requires-Dist: plotly

{lt_tensor-0.0.1a20 → lt_tensor-0.0.1a21}/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.2a3
+lt-utils==0.0.2
 librosa==0.11.*
 einops
 plotly

{lt_tensor-0.0.1a20 → lt_tensor-0.0.1a21}/setup.py

@@ -4,7 +4,7 @@ with open("README.md", "r", encoding="utf-8") as f:
     long_description = f.read()
 
 setup(
-    version="0.0.1a20",
+    version="0.0.1a21",
     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.2a3",
+        "lt-utils==0.0.2",
         "librosa==0.11.*",
         "einops",
         "plotly",