unifiedefficientloader 0.5.3__tar.gz → 0.5.4__tar.gz

{unifiedefficientloader-0.5.3 → unifiedefficientloader-0.5.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: unifiedefficientloader
-Version: 0.5.3
+Version: 0.5.4
 Summary: A unified interface for memory efficient per tensor loading of safetensors files as raw bytes from offset, handling CPU/GPU pinned transfers, and converting between tensors and dicts.
 Author: silveroxides
 License: MIT License
@@ -33,7 +33,7 @@ Classifier: Operating System :: POSIX :: Linux
 Requires-Python: >=3.9
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: comfy-aimdo>=0.4.5
+Requires-Dist: comfy-aimdo>=0.5.1
 Provides-Extra: torch
 Requires-Dist: torch; extra == "torch"
 Provides-Extra: safetensors

{unifiedefficientloader-0.5.3 → unifiedefficientloader-0.5.4}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "unifiedefficientloader"
-version = "0.5.3"
+version = "0.5.4"
 description = "A unified interface for memory efficient per tensor loading of safetensors files as raw bytes from offset, handling CPU/GPU pinned transfers, and converting between tensors and dicts."
 readme = "README.md"
 authors = [
@@ -19,7 +19,7 @@ classifiers = [
     "Operating System :: POSIX :: Linux",
 ]
 requires-python = ">=3.9"
-dependencies = ["comfy-aimdo>=0.4.5"]
+dependencies = ["comfy-aimdo>=0.5.1"]
 
 [project.optional-dependencies]
 torch = ["torch"]

{unifiedefficientloader-0.5.3 → unifiedefficientloader-0.5.4}/tests/test_mmap.py

@@ -106,6 +106,29 @@ def test_mmap_get_tensor_values(sample_file):
             assert torch.equal(t.clone(), expected), f"{key}: value mismatch"
 
 
+@uel_required
+def test_mmap_exposes_file_handle(sample_file):
+    """The compatibility wrapper exposes comfy-aimdo's file handle API."""
+    path, _ = sample_file
+
+    with UnifiedSafetensorsLoader(path, low_memory=True, use_mmap=True) as loader:
+        file_handle = loader._mmap.get_file_handle()
+
+    assert isinstance(file_handle, int)
+    assert file_handle > 0
+
+
+def test_control_init_accepts_aimdo_051_options():
+    """The control wrapper forwards comfy-aimdo 0.5.1 init options."""
+    from unifiedefficientloader.uel import control
+
+    assert control.init(
+        implementation=None,
+        simple_vram_headroom=None,
+        nvml_pressure=False,
+    )
+
+
 @uel_required
 def test_mmap_get_tensor_mmap_ref_attached(sample_file):
     """

{unifiedefficientloader-0.5.3 → unifiedefficientloader-0.5.4}/unifiedefficientloader/uel/model_mmap.py

@@ -23,6 +23,9 @@ def _setup_argtypes(lib):
     lib.model_mmap_get.argtypes = [ctypes.c_void_p]
     lib.model_mmap_get.restype = ctypes.c_void_p
 
+    lib.model_mmap_get_file_handle.argtypes = [ctypes.c_void_p]
+    lib.model_mmap_get_file_handle.restype = ctypes.c_uint64
+
     lib.model_mmap_bounce.argtypes = [ctypes.c_void_p]
     lib.model_mmap_bounce.restype = ctypes.c_bool
 
@@ -53,6 +56,9 @@ class ModelMMAP:
     def get(self):
         return _ctrl.lib.model_mmap_get(self.state)
 
+    def get_file_handle(self):
+        return int(_ctrl.lib.model_mmap_get_file_handle(self.state))
+
     def bounce(self):
         return bool(_ctrl.lib.model_mmap_bounce(self.state))
 

{unifiedefficientloader-0.5.3 → unifiedefficientloader-0.5.4}/unifiedefficientloader.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: unifiedefficientloader
-Version: 0.5.3
+Version: 0.5.4
 Summary: A unified interface for memory efficient per tensor loading of safetensors files as raw bytes from offset, handling CPU/GPU pinned transfers, and converting between tensors and dicts.
 Author: silveroxides
 License: MIT License
@@ -33,7 +33,7 @@ Classifier: Operating System :: POSIX :: Linux
 Requires-Python: >=3.9
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: comfy-aimdo>=0.4.5
+Requires-Dist: comfy-aimdo>=0.5.1
 Provides-Extra: torch
 Requires-Dist: torch; extra == "torch"
 Provides-Extra: safetensors

{unifiedefficientloader-0.5.3 → unifiedefficientloader-0.5.4}/unifiedefficientloader.egg-info/requires.txt

@@ -1,4 +1,4 @@
-comfy-aimdo>=0.4.5
+comfy-aimdo>=0.5.1
 
 [all]
 torch