unifiedefficientloader 0.5.0__tar.gz → 0.5.2__tar.gz

{unifiedefficientloader-0.5.0 → unifiedefficientloader-0.5.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: unifiedefficientloader
-Version: 0.5.0
+Version: 0.5.2
 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.3
+Requires-Dist: comfy-aimdo>=0.4.5
 Provides-Extra: torch
 Requires-Dist: torch; extra == "torch"
 Provides-Extra: safetensors

{unifiedefficientloader-0.5.0 → unifiedefficientloader-0.5.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "unifiedefficientloader"
-version = "0.5.0"
+version = "0.5.2"
 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.3"]
+dependencies = ["comfy-aimdo>=0.4.5"]
 
 [project.optional-dependencies]
 torch = ["torch"]

{unifiedefficientloader-0.5.0 → unifiedefficientloader-0.5.2}/unifiedefficientloader/uel/control.py

@@ -16,7 +16,7 @@ _original_init_device = _ctrl.init_device
 
 def _patch_lib_refs():
     """Propagate control.lib to submodules that captured it at import time."""
-    for mod_name in ("comfy_aimdo.model_mmap", "comfy_aimdo.host_buffer", "comfy_aimdo.model_vbar"):
+    for mod_name in ("comfy_aimdo.model_mmap", "comfy_aimdo.host_buffer", "comfy_aimdo.model_vbar", "comfy_aimdo.vram_buffer"):
         mod = sys.modules.get(mod_name)
         if mod is not None and hasattr(mod, "lib"):
             mod.lib = _ctrl.lib

{unifiedefficientloader-0.5.0 → unifiedefficientloader-0.5.2}/unifiedefficientloader.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: unifiedefficientloader
-Version: 0.5.0
+Version: 0.5.2
 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.3
+Requires-Dist: comfy-aimdo>=0.4.5
 Provides-Extra: torch
 Requires-Dist: torch; extra == "torch"
 Provides-Extra: safetensors

{unifiedefficientloader-0.5.0 → unifiedefficientloader-0.5.2}/unifiedefficientloader.egg-info/requires.txt

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