comfy-env 0.0.48__py3-none-any.whl → 0.0.49__py3-none-any.whl

comfy_env/decorator.py

@@ -178,7 +178,10 @@ def _get_or_create_worker(config: WorkerConfig, log_fn: Callable):
             # Same Python - use TorchMPWorker (fast, zero-copy)
             from .workers import TorchMPWorker
             log_fn(f"Creating TorchMPWorker (same Python, zero-copy tensors)")
-            worker = TorchMPWorker(name=config.env_name, sys_path=config.sys_path)
+            worker = TorchMPWorker(
+                name=config.env_name,
+                sys_path=config.sys_path,
+            )
         else:
             # Different Python - use PersistentVenvWorker
             from .workers.venv import PersistentVenvWorker

comfy_env/templates/comfy-env-instructions.txt

@@ -0,0 +1,82 @@
+comfy-env Setup Instructions
+============================
+
+comfy-env handles CUDA wheel installation and process isolation for ComfyUI
+custom nodes.
+
+QUICK START
+-----------
+1. Install comfy-env:
+   pip install comfy-env
+
+2. Initialize config in your node directory:
+   comfy-env init
+
+3. Edit comfy-env.toml to add your dependencies
+
+4. Add to your __init__.py (at the top, before other imports):
+   from comfy_env import install
+   install()
+
+5. Test locally:
+   comfy-env install --dry-run  # Preview what will be installed
+   comfy-env install            # Actually install
+   comfy-env doctor             # Verify all packages work
+
+
+COMMON USE CASES
+----------------
+
+Case 1: Just need CUDA packages (nvdiffrast, pytorch3d, etc.)
+  - Add packages to [cuda] section
+  - Call install() in your __init__.py
+
+Case 2: Need process isolation (conflicting dependencies)
+  - Define an isolated environment section like [myenv]
+  - Use @isolated(env="myenv") decorator on your node class
+
+Case 3: Need system packages (apt)
+  - Add to [system] linux = ["package1", "package2"]
+
+
+CLI COMMANDS
+------------
+  comfy-env init           Create comfy-env.toml template
+  comfy-env install        Install dependencies from config
+  comfy-env install --dry-run  Preview without installing
+  comfy-env info           Show detected environment (Python, CUDA, PyTorch)
+  comfy-env resolve PKG    Show resolved wheel URL for a package
+  comfy-env doctor         Verify installation
+  comfy-env list-packages  Show all packages in built-in registry
+
+
+PROCESS ISOLATION
+-----------------
+For nodes that need isolated dependencies:
+
+  from comfy_env import isolated
+
+  @isolated(env="myenv")
+  class MyNode:
+      FUNCTION = "process"
+      RETURN_TYPES = ("IMAGE",)
+
+      def process(self, image):
+          # This runs in isolated subprocess
+          import conflicting_lib
+          return (result,)
+
+The decorator:
+  - Runs your node's FUNCTION method in a separate Python process
+  - Transfers tensors efficiently via PyTorch IPC (zero-copy for CUDA)
+  - Uses the venv defined in your comfy-env.toml [myenv] section
+
+
+TROUBLESHOOTING
+---------------
+- "Package X not found in registry": Add custom wheel URL to [wheel_sources]
+- "CUDA not detected": Ensure PyTorch with CUDA is installed in ComfyUI
+- "Worker failed to connect": Check the isolated env was set up correctly
+- Import errors: Run `comfy-env doctor` to verify packages
+
+For more help: https://github.com/PozzettiAndrea/comfy-env

comfy_env/templates/comfy-env.toml

@@ -0,0 +1,211 @@
+# =============================================================================
+# comfy-env.toml - Environment configuration for ComfyUI custom nodes
+# Documentation: https://github.com/PozzettiAndrea/comfy-env
+# =============================================================================
+#
+# This file configures dependencies for your ComfyUI custom node.
+# comfy-env handles two main challenges:
+#   1. Installing CUDA packages that require compilation (nvdiffrast, pytorch3d, etc.)
+#   2. Running nodes in isolated environments with their own dependencies
+#
+# Quick start:
+#   pip install comfy-env
+#   comfy-env init          # Creates this template
+#   comfy-env install       # Installs dependencies
+#   comfy-env doctor        # Verifies installation
+
+
+# =============================================================================
+# SYSTEM PACKAGES (optional)
+# =============================================================================
+# System-level packages installed via apt (Linux) before Python packages.
+# Useful for packages that need system libraries to compile or run.
+
+[system]
+linux = []
+# Examples:
+# linux = ["libgl1", "libopengl0"]     # For OpenGL rendering (nvdiffrast)
+# linux = ["python3-dev"]               # For packages that compile C extensions
+# linux = ["ffmpeg"]                    # For video processing nodes
+
+
+# =============================================================================
+# CUDA PACKAGES
+# =============================================================================
+# CUDA packages from the built-in registry. These are pre-compiled wheels
+# that install without needing a compiler. Run `comfy-env list-packages`
+# to see all available packages.
+
+[cuda]
+# Use exact package names from `comfy-env list-packages` (hyphens vs underscores matter!)
+# nvdiffrast = "0.4.0"       # NVIDIA differentiable rasterizer
+# pytorch3d = "0.7.9"        # PyTorch3D - 3D deep learning
+# gsplat = "1.5.3"           # Gaussian splatting rasterization
+# torch-scatter = "2.1.2"    # PyG scatter operations
+# torch-cluster = "1.6.3"    # PyG clustering algorithms
+# flash-attn = "2.7.4"       # Flash Attention (Linux x86_64 only)
+# spconv = "2.3.8"           # Sparse convolution library
+# sageattention = "2.2.0"    # SageAttention - faster than FlashAttention
+# dpvo_cuda = "0.0.0"        # DPVO CUDA extensions
+
+
+# =============================================================================
+# REGULAR PACKAGES
+# =============================================================================
+# Standard pip packages (no special CUDA handling needed).
+
+[packages]
+requirements = []
+# Examples:
+# requirements = [
+#     "transformers>=4.56",
+#     "pillow",
+#     "opencv-python-headless",
+#     "trimesh",
+# ]
+
+
+# =============================================================================
+# CUSTOM WHEEL SOURCES (optional)
+# =============================================================================
+# Override built-in wheel URLs or add packages not in the registry.
+# Template variables: {version}, {cuda_short}, {torch_mm}, {py_tag}, {platform}
+
+[wheel_sources]
+# my-custom-package = "https://my-server.com/my-package-{version}+cu{cuda_short}-{py_tag}-{platform}.whl"
+
+
+# =============================================================================
+# NODE DEPENDENCIES (optional)
+# =============================================================================
+# Other ComfyUI custom nodes this node depends on.
+
+[node_reqs]
+# VideoHelperSuite = "Kosinkadink/ComfyUI-VideoHelperSuite"
+# ComfyUI-Impact-Pack = "ltdrdata/ComfyUI-Impact-Pack"
+
+
+# =============================================================================
+# EXTERNAL TOOLS (optional)
+# =============================================================================
+# External applications required by the node.
+
+[tools]
+# blender = "4.2"
+
+
+# #############################################################################
+#
+#                       PROCESS ISOLATION (ADVANCED)
+#
+# #############################################################################
+#
+# For nodes that need completely isolated dependencies (different PyTorch
+# version, conflicting native libraries, etc.), define isolated environments.
+#
+# How it works:
+#   1. Define an environment below with its own Python/CUDA/packages
+#   2. Use the @isolated decorator on your node class
+#   3. The node runs in a separate subprocess with its own venv
+#
+# Example in __init__.py:
+#
+#     from comfy_env import isolated
+#
+#     @isolated(env="myenv")
+#     class MyIsolatedNode:
+#         FUNCTION = "process"
+#         RETURN_TYPES = ("IMAGE",)
+#
+#         def process(self, image):
+#             # This runs in isolated subprocess with its own dependencies
+#             import conflicting_package
+#             return (result,)
+#
+# =============================================================================
+
+
+# -----------------------------------------------------------------------------
+# Example: Simple isolation (same Python, different process)
+# -----------------------------------------------------------------------------
+# Use when you just need process isolation but no venv (fastest option).
+# In your code: @isolated(env="simple", same_venv=True)
+
+# [simple]
+# python = "3.11"
+# cuda_version = "auto"
+# pytorch_version = "auto"
+
+
+# -----------------------------------------------------------------------------
+# Example: Full venv isolation
+# -----------------------------------------------------------------------------
+# Use when you need completely different dependencies.
+# Creates a separate virtual environment with its own packages.
+
+# [myenv]
+# python = "3.10"
+# cuda_version = "12.8"
+# pytorch_version = "2.8.0"
+#
+# [myenv.cuda]
+# nvdiffrast = "0.4.0"
+# pytorch3d = "0.7.9"
+#
+# [myenv.packages]
+# requirements = ["trimesh", "scipy"]
+
+
+# -----------------------------------------------------------------------------
+# Example: Multiple isolated environments
+# -----------------------------------------------------------------------------
+# Some nodes need multiple different environments for different operations.
+
+# [env-preprocessing]
+# python = "3.11"
+# cuda_version = "12.8"
+# pytorch_version = "2.8.0"
+#
+# [env-preprocessing.packages]
+# requirements = ["opencv-python-headless", "pillow"]
+
+# [env-inference]
+# python = "3.10"
+# cuda_version = "12.4"
+# pytorch_version = "2.5.1"
+#
+# [env-inference.cuda]
+# torch-scatter = "2.1.2"  # Use exact name from `comfy-env list-packages`
+
+
+# -----------------------------------------------------------------------------
+# Example: Platform-specific packages
+# -----------------------------------------------------------------------------
+# Different packages for Windows vs Linux.
+
+# [crossplatform]
+# python = "3.11"
+# cuda_version = "auto"
+#
+# [crossplatform.packages]
+# requirements = ["numpy", "pillow"]
+#
+# [crossplatform.packages.windows]
+# requirements = ["pywin32"]
+#
+# [crossplatform.packages.linux]
+# requirements = ["python-xlib"]
+
+
+# -----------------------------------------------------------------------------
+# Example: Conda packages (uses pixi backend)
+# -----------------------------------------------------------------------------
+# For packages that are easier to install via conda (e.g., CUDA toolkit).
+
+# [blender-env]
+# python = "3.11"
+# cuda_version = "12.8"
+#
+# [blender-env.conda]
+# channels = ["conda-forge", "nvidia"]
+# packages = ["cuda-toolkit=12.8", "bpy"]

comfy_env/wheel_sources.yml

@@ -107,7 +107,7 @@ packages:
     default_version: "2.2.0"
     description: SageAttention - 2-5x faster than FlashAttention with quantized kernels
 
-  dpvo_cuda:
+  dpvo-cuda:
     wheel_template: "https://github.com/PozzettiAndrea/cuda-wheels/releases/download/dpvo_cuda-latest/dpvo_cuda-{version}%2Bcu{cuda_short}torch{torch_mm}-{py_tag}-{py_tag}-{platform}.whl"
     default_version: "0.0.0"
     description: DPVO CUDA extensions (cuda_corr, cuda_ba, lietorch_backends) - torch 2.4 only

comfy_env/workers/torch_mp.py

@@ -50,8 +50,12 @@ def _worker_loop(queue_in, queue_out, sys_path_additions=None):
     - _SHUTDOWN: Shutdown the worker
 
     Runs until receiving _SHUTDOWN sentinel.
+
+    Args:
+        queue_in: Input queue for receiving work items
+        queue_out: Output queue for sending results
+        sys_path_additions: Paths to add to sys.path
     """
-    import importlib
     import os
     import sys
     from pathlib import Path

comfy_env/workers/venv.py

@@ -912,7 +912,9 @@ class PersistentVenvWorker(Worker):
         self._transport = SocketTransport(client_sock)
 
         # Send config
-        config = {"sys_paths": all_sys_path}
+        config = {
+            "sys_paths": all_sys_path,
+        }
         self._transport.send(config)
 
         # Wait for ready signal

{comfy_env-0.0.48.dist-info → comfy_env-0.0.49.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: comfy-env
-Version: 0.0.48
+Version: 0.0.49
 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.48.dist-info → comfy_env-0.0.49.dist-info}/RECORD

@@ -1,6 +1,6 @@
 comfy_env/__init__.py,sha256=OQJFNjmArjLcgrfHAFxgDJQFH_IhxibqMXbU5bu_j9Q,3822
 comfy_env/cli.py,sha256=Pjzb-jsoH67lyHxmBFOWvasWX01eHFE_BEjUk1l-uEo,14509
-comfy_env/decorator.py,sha256=6JCKwLHaZtOLVDexs_gh_-NtS2ZK0V7nGCPqkyeYEAA,16688
+comfy_env/decorator.py,sha256=sCb3DQIYYdt8fuMiJnmNfU8oLawbfjMMR2Hf0YMT4B0,16738
 comfy_env/errors.py,sha256=8hN8NDlo8oBUdapc-eT3ZluigI5VBzfqsSBvQdfWlz4,9943
 comfy_env/install.py,sha256=m4NKlfCcQGI5xzVRjHw3ep-lWbqx5kE1e21sUUZ2Leo,17528
 comfy_env/nodes.py,sha256=CWUe35jU5SKk4ur-SddZePdqWgxJDlxGhpcJiu5pAK4,4354
@@ -30,15 +30,17 @@ comfy_env/stubs/folder_paths.py,sha256=K90J34EG6LD4eZP8YG-xMeBmqwpp_wA8E92DKMXd1
 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/templates/comfy-env-instructions.txt,sha256=pXxlyOxWKq8NYtxIsa9wD8Y3qzC8JUWguTPiV9dB0Mw,2481
+comfy_env/templates/comfy-env.toml,sha256=1CyphZu9DVc-EKzkLiDCnNzY9AWH5s_Kp9o4s2Fnw9g,7441
 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
 comfy_env/workers/tensor_utils.py,sha256=TCuOAjJymrSbkgfyvcKtQ_KbVWTqSwP9VH_bCaFLLq8,6409
-comfy_env/workers/torch_mp.py,sha256=4YSNPn7hALrvMVbkO4RkTeFTcc0lhfLMk5QTWjY4PHw,22134
-comfy_env/workers/venv.py,sha256=9xP3Y_ZoegMMCnBuRrd-mlIEKRF4oE01zL_8bngRuCk,38916
-comfy_env/wheel_sources.yml,sha256=K5dksy21YcT7QdFlVDkKF4Rv9ZCjHaWhQgoEhdSyAOI,8156
-comfy_env-0.0.48.dist-info/METADATA,sha256=a8hWh2c1hJZt2WkfR4giYSjl86bBTDuXYWB0Mn2z-8E,7138
-comfy_env-0.0.48.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-comfy_env-0.0.48.dist-info/entry_points.txt,sha256=J4fXeqgxU_YenuW_Zxn_pEL7J-3R0--b6MS5t0QmAr0,49
-comfy_env-0.0.48.dist-info/licenses/LICENSE,sha256=E68QZMMpW4P2YKstTZ3QU54HRQO8ecew09XZ4_Vn870,1093
-comfy_env-0.0.48.dist-info/RECORD,,
+comfy_env/workers/torch_mp.py,sha256=TnsCoBHEJBXEoBkx7WiCd9tBAlzFtMOw1dk_7_zGJZY,22288
+comfy_env/workers/venv.py,sha256=kAaADoUJpKItZoG73YZhn0ufqzHmDgU1xeSHNCiZ77I,38941
+comfy_env/wheel_sources.yml,sha256=uU0YJmWaiLAicQNN9VYS8PZevlP2NOH6mBUE294dvAo,8156
+comfy_env-0.0.49.dist-info/METADATA,sha256=Ve0XcboM1LLVeXVFYTbBDMxIcefaYiYnC8Nf84inrhQ,7138
+comfy_env-0.0.49.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+comfy_env-0.0.49.dist-info/entry_points.txt,sha256=J4fXeqgxU_YenuW_Zxn_pEL7J-3R0--b6MS5t0QmAr0,49
+comfy_env-0.0.49.dist-info/licenses/LICENSE,sha256=E68QZMMpW4P2YKstTZ3QU54HRQO8ecew09XZ4_Vn870,1093
+comfy_env-0.0.49.dist-info/RECORD,,