mmgp 3.4.3__tar.gz → 3.4.4__tar.gz

{mmgp-3.4.3/src/mmgp.egg-info → mmgp-3.4.4}/PKG-INFO

@@ -1,10 +1,8 @@
 Metadata-Version: 2.4
 Name: mmgp
-Version: 3.4.3
+Version: 3.4.4
 Summary: Memory Management for the GPU Poor
 Author-email: deepbeepmeep <deepbeepmeep@yahoo.com>
-License:                     GNU GENERAL PUBLIC LICENSE
-                               Version 3, 29 June 2007
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE.md
@@ -17,7 +15,7 @@ Dynamic: license-file
 
 
 <p align="center">
-  <H2>Memory Management 3.4.3 for the GPU Poor by DeepBeepMeep</H2>	
+  <H2>Memory Management 3.4.4 for the GPU Poor by DeepBeepMeep</H2>	
 </p>
 
 

{mmgp-3.4.3 → mmgp-3.4.4}/README.md

@@ -1,6 +1,6 @@
 
 <p align="center">
-  <H2>Memory Management 3.4.3 for the GPU Poor by DeepBeepMeep</H2>	
+  <H2>Memory Management 3.4.4 for the GPU Poor by DeepBeepMeep</H2>	
 </p>
 
 

{mmgp-3.4.3 → mmgp-3.4.4}/pyproject.toml

@@ -1,13 +1,13 @@
 [project]
 name = "mmgp"
-version = "3.4.3"
+version = "3.4.4"
 authors = [
   { name = "deepbeepmeep", email = "deepbeepmeep@yahoo.com" },
 ]
 description = "Memory Management for the GPU Poor"
 readme = "README.md"
 requires-python = ">=3.10"
-license = { file = "LICENSE.md" }
+license-files = ["LICENSE.md"]
 dependencies = [
   "torch >= 2.1.0",
   "optimum-quanto",

{mmgp-3.4.3 → mmgp-3.4.4}/src/mmgp/safetensors2.py

@@ -1,4 +1,4 @@
-# ------------------ Safetensors2 1.1 by DeepBeepMeep (mmgp)------------------
+# ------------------ Safetensors2 1.3 by DeepBeepMeep (mmgp)------------------
 #
 # This module entirely written in Python is a replacement for the safetensor library which requires much less RAM to load models.
 # It can be conveniently used to keep a low RAM consumption when handling  transit data (for instance when quantizing or transferring tensors to reserver RAM)
@@ -450,7 +450,17 @@ class _SafeTensorLoader:
     def __exit__(self, exc_type, exc_val, exc_tb) -> None:
         """Clean up resources"""        
         self.close()
-        
+
+    def get_tensor(self, name):
+        if self.sft == None:
+            self.__enter__()
+        return self.sft.get_tensor(name)
+
+    def get_slice(self, name):
+        if self.sft == None:
+            self.__enter__()
+        return self.sft.get_slice(name)
+
     def close(self) -> None:
         if self.sft != None:
             self.sft._free_resources()

{mmgp-3.4.3 → mmgp-3.4.4/src/mmgp.egg-info}/PKG-INFO

@@ -1,10 +1,8 @@
 Metadata-Version: 2.4
 Name: mmgp
-Version: 3.4.3
+Version: 3.4.4
 Summary: Memory Management for the GPU Poor
 Author-email: deepbeepmeep <deepbeepmeep@yahoo.com>
-License:                     GNU GENERAL PUBLIC LICENSE
-                               Version 3, 29 June 2007
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE.md
@@ -17,7 +15,7 @@ Dynamic: license-file
 
 
 <p align="center">
-  <H2>Memory Management 3.4.3 for the GPU Poor by DeepBeepMeep</H2>	
+  <H2>Memory Management 3.4.4 for the GPU Poor by DeepBeepMeep</H2>	
 </p>