comfy-env 0.0.39__py3-none-any.whl → 0.0.40__py3-none-any.whl

comfy_env/stubs/comfy/__init__.py

@@ -0,0 +1,6 @@
+"""
+Comfy stubs package for isolated worker processes.
+
+Provides minimal implementations of commonly used comfy modules
+without requiring the full ComfyUI installation.
+"""

comfy_env/stubs/comfy/model_management.py

@@ -0,0 +1,58 @@
+"""
+Stub for comfy.model_management in isolated worker processes.
+
+Provides device detection and memory management functions without
+requiring the full ComfyUI installation.
+"""
+
+import torch
+
+
+def get_torch_device():
+    """Return the best available torch device."""
+    if torch.cuda.is_available():
+        return torch.device("cuda")
+    elif hasattr(torch.backends, "mps") and torch.backends.mps.is_available():
+        return torch.device("mps")
+    return torch.device("cpu")
+
+
+def get_free_memory(device=None, torch_free_too=False):
+    """Return free VRAM in bytes."""
+    if device is None:
+        device = get_torch_device()
+    if device.type == "cuda":
+        free, total = torch.cuda.mem_get_info(device)
+        return free
+    return 0
+
+
+def get_total_memory(device=None, torch_total_too=False):
+    """Return total VRAM in bytes."""
+    if device is None:
+        device = get_torch_device()
+    if device.type == "cuda":
+        free, total = torch.cuda.mem_get_info(device)
+        return total
+    return 0
+
+
+def soft_empty_cache(force=False):
+    """Clear CUDA cache."""
+    if torch.cuda.is_available():
+        torch.cuda.empty_cache()
+
+
+def unload_all_models():
+    """No-op in isolated worker - models managed by the node itself."""
+    pass
+
+
+def interrupt_current_processing(value=True):
+    """No-op in isolated worker."""
+    pass
+
+
+def processing_interrupted():
+    """Always returns False in isolated worker."""
+    return False

comfy_env/stubs/comfy/utils.py

@@ -0,0 +1,29 @@
+"""
+Stub for comfy.utils in isolated worker processes.
+
+Provides utility classes like ProgressBar without requiring
+the full ComfyUI installation.
+"""
+
+
+class ProgressBar:
+    """
+    No-op progress bar for isolated workers.
+
+    In isolated subprocess, we can't update the main ComfyUI progress bar,
+    so this just tracks progress internally. Nodes can still use the same API.
+    """
+
+    def __init__(self, total):
+        self.total = total
+        self.current = 0
+
+    def update(self, value):
+        """Increment progress by value."""
+        self.current += value
+
+    def update_absolute(self, value, total=None, preview=None):
+        """Set progress to absolute value."""
+        self.current = value
+        if total is not None:
+            self.total = total

{comfy_env-0.0.39.dist-info → comfy_env-0.0.40.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: comfy-env
-Version: 0.0.39
+Version: 0.0.40
 Summary: Environment management for ComfyUI custom nodes - CUDA wheel resolution and process isolation
 Project-URL: Homepage, https://github.com/PozzettiAndrea/comfy-env
 Project-URL: Repository, https://github.com/PozzettiAndrea/comfy-env

{comfy_env-0.0.39.dist-info → comfy_env-0.0.40.dist-info}/RECORD

@@ -28,6 +28,9 @@ comfy_env/ipc/transport.py,sha256=XQlRcfQsd4nd909KIYnZKvsS3ksGpGjyVucn8jvmLIU,96
 comfy_env/ipc/worker.py,sha256=oxTLF9xXrl8CRx_JVNBdkxZh35NuzfkdxhaUtUuXogs,6661
 comfy_env/stubs/__init__.py,sha256=jMeWEKY30y8QqYX9AUyuZbmm607erQTc4N7YaDoAH00,38
 comfy_env/stubs/folder_paths.py,sha256=K90J34EG6LD4eZP8YG-xMeBmqwpp_wA8E92DKMXd1GQ,2189
+comfy_env/stubs/comfy/__init__.py,sha256=-y4L6gX21vrI2V8MvNaMeHOcAn5kUNK3jUyLvtXRmJQ,173
+comfy_env/stubs/comfy/model_management.py,sha256=Khx8Qa3NutKPLTn9oSM3VLeATUOg1fe4QCjxdxXd6eE,1462
+comfy_env/stubs/comfy/utils.py,sha256=s3t_KLj_-w1Uj3A3iAy69wIk4Ggklojw5hsDNb69Pcc,776
 comfy_env/workers/__init__.py,sha256=IKZwOvrWOGqBLDUIFAalg4CdqzJ_YnAdxo2Ha7gZTJ0,1467
 comfy_env/workers/base.py,sha256=ZILYXlvGCWuCZXmjKqfG8VeD19ihdYaASdlbasl2BMo,2312
 comfy_env/workers/pool.py,sha256=MtjeOWfvHSCockq8j1gfnxIl-t01GSB79T5N4YB82Lg,6956
@@ -35,8 +38,8 @@ comfy_env/workers/tensor_utils.py,sha256=TCuOAjJymrSbkgfyvcKtQ_KbVWTqSwP9VH_bCaF
 comfy_env/workers/torch_mp.py,sha256=4YSNPn7hALrvMVbkO4RkTeFTcc0lhfLMk5QTWjY4PHw,22134
 comfy_env/workers/venv.py,sha256=PmsVOu5i89tBYkGRupo2bjOLPBmk06q4GNUwDWsd9F8,32088
 comfy_env/wheel_sources.yml,sha256=IEyi3JnySo4sXTRkYR0VjtCdXaCsF5Q5c0nU0S9LC8I,9143
-comfy_env-0.0.39.dist-info/METADATA,sha256=S5u6Cq12jtVNGzgeXTSjN3Cr4CkfVGHExYeDvJGJ0EA,5400
-comfy_env-0.0.39.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-comfy_env-0.0.39.dist-info/entry_points.txt,sha256=J4fXeqgxU_YenuW_Zxn_pEL7J-3R0--b6MS5t0QmAr0,49
-comfy_env-0.0.39.dist-info/licenses/LICENSE,sha256=E68QZMMpW4P2YKstTZ3QU54HRQO8ecew09XZ4_Vn870,1093
-comfy_env-0.0.39.dist-info/RECORD,,
+comfy_env-0.0.40.dist-info/METADATA,sha256=eJYprZ_ULUXc7vWu10nqRaMntOuerWRSJpBYf4zXUwc,5400
+comfy_env-0.0.40.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+comfy_env-0.0.40.dist-info/entry_points.txt,sha256=J4fXeqgxU_YenuW_Zxn_pEL7J-3R0--b6MS5t0QmAr0,49
+comfy_env-0.0.40.dist-info/licenses/LICENSE,sha256=E68QZMMpW4P2YKstTZ3QU54HRQO8ecew09XZ4_Vn870,1093
+comfy_env-0.0.40.dist-info/RECORD,,