comfy-env 0.0.25__tar.gz → 0.0.27__tar.gz

{comfy_env-0.0.25 → comfy_env-0.0.27}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: comfy-env
-Version: 0.0.25
+Version: 0.0.27
 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.25 → comfy_env-0.0.27}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "comfy-env"
-version = "0.0.25"
+version = "0.0.27"
 description = "Environment management for ComfyUI custom nodes - CUDA wheel resolution and process isolation"
 readme = "README.md"
 license = {text = "MIT"}

{comfy_env-0.0.25 → comfy_env-0.0.27}/src/comfy_env/install.py

@@ -24,7 +24,7 @@ import sys
 from pathlib import Path
 from typing import Any, Callable, Dict, List, Optional, Set, Union
 
-from .env.config import IsolatedEnv, NodeReq, SystemConfig
+from .env.config import IsolatedEnv, LocalConfig, NodeReq, SystemConfig
 from .env.config_file import load_config, discover_config
 from .env.manager import IsolatedEnvManager
 from .errors import CUDANotFoundError, DependencyError, InstallError, WheelNotFoundError
@@ -254,6 +254,9 @@ def install(
         return _install_isolated(env_config, node_dir, log, dry_run)
     elif env_config:
         return _install_inplace(env_config, node_dir, log, dry_run, verify_wheels)
+    elif full_config.has_local:
+        # Handle [local.cuda] and [local.packages] without isolated env
+        return _install_local(full_config.local, node_dir, log, dry_run)
     else:
         return True
 
@@ -353,6 +356,63 @@ def _install_inplace(
     return True
 
 
+def _install_local(
+    local_config: LocalConfig,
+    node_dir: Path,
+    log: Callable[[str], None],
+    dry_run: bool,
+) -> bool:
+    """Install local packages into current environment (no isolated venv)."""
+    log("Installing local packages into host environment")
+
+    # Install MSVC runtime on Windows (required for CUDA/PyTorch native extensions)
+    if sys.platform == "win32":
+        log("Installing MSVC runtime for Windows...")
+        if not dry_run:
+            _pip_install(["msvc-runtime"], no_deps=False, log=log)
+
+    # Detect runtime environment
+    env = RuntimeEnv.detect()
+    log(f"Detected environment: {env}")
+
+    # Check CUDA requirement
+    if not env.cuda_version and local_config.cuda_packages:
+        raise CUDANotFoundError(package=", ".join(local_config.cuda_packages.keys()))
+
+    # Convert cuda_packages dict to list of specs
+    cuda_packages = []
+    for pkg, ver in local_config.cuda_packages.items():
+        if ver:
+            cuda_packages.append(f"{pkg}=={ver}")
+        else:
+            cuda_packages.append(pkg)
+
+    if dry_run:
+        log("\nDry run - would install:")
+        for pkg in cuda_packages:
+            log(f"  {pkg}")
+        if local_config.requirements:
+            log("  Regular packages:")
+            for pkg in local_config.requirements:
+                log(f"    {pkg}")
+        return True
+
+    # Install CUDA packages
+    if cuda_packages:
+        log(f"\nInstalling {len(cuda_packages)} CUDA packages...")
+        for req in cuda_packages:
+            package, version = parse_wheel_requirement(req)
+            _install_cuda_package(package, version, env, [], log)
+
+    # Install regular packages
+    if local_config.requirements:
+        log(f"\nInstalling {len(local_config.requirements)} regular packages...")
+        _pip_install(local_config.requirements, no_deps=False, log=log)
+
+    log("\nLocal installation complete!")
+    return True
+
+
 def _get_install_info(
     package: str,
     version: Optional[str],

{comfy_env-0.0.25 → comfy_env-0.0.27}/src/comfy_env/wheel_sources.yml

@@ -58,38 +58,48 @@ packages:
     description: PyTorch3D - 3D deep learning library
 
   # ===========================================================================
-  # PozzettiAndrea wheel repos (GitHub Pages indexes)
+  # PozzettiAndrea cuda-wheels (unified PEP 503 index)
+  # https://pozzettiandrea.github.io/cuda-wheels
   # ===========================================================================
   nvdiffrast:
-    method: github_index
-    index_url: "https://pozzettiandrea.github.io/nvdiffrast-full-wheels/cu{cuda_short}-torch{torch_short}/"
+    method: index
+    index_url: "https://pozzettiandrea.github.io/cuda-wheels"
     description: NVIDIA differentiable rasterizer
 
   cumesh:
-    method: github_index
-    index_url: "https://pozzettiandrea.github.io/cumesh-wheels/cu{cuda_short}-torch{torch_short}/"
+    method: index
+    index_url: "https://pozzettiandrea.github.io/cuda-wheels"
     description: CUDA-accelerated mesh utilities
 
   o_voxel:
-    method: github_index
-    index_url: "https://pozzettiandrea.github.io/ovoxel-wheels/cu{cuda_short}-torch{torch_short}/"
+    method: index
+    index_url: "https://pozzettiandrea.github.io/cuda-wheels"
     description: O-Voxel CUDA extension for TRELLIS
 
   flex_gemm:
-    method: github_index
-    index_url: "https://pozzettiandrea.github.io/flexgemm-wheels/cu{cuda_short}-torch{torch_short}/"
+    method: index
+    index_url: "https://pozzettiandrea.github.io/cuda-wheels"
     description: Flexible GEMM operations
 
   nvdiffrec_render:
-    method: github_release
+    method: index
+    index_url: "https://pozzettiandrea.github.io/cuda-wheels"
     description: NVDiffRec rendering utilities
-    sources:
-      - name: PozzettiAndrea
-        url_template: "https://github.com/PozzettiAndrea/nvdiffrec_render-wheels/releases/download/cu{cuda_short}-torch{torch_short}/nvdiffrec_render-{version}%2Bcu{cuda_short}torch{torch_mm}-{py_tag}-{py_tag}-linux_x86_64.whl"
-        platforms: [linux_x86_64]
-      - name: PozzettiAndrea-windows
-        url_template: "https://github.com/PozzettiAndrea/nvdiffrec_render-wheels/releases/download/cu{cuda_short}-torch{torch_short}/nvdiffrec_render-{version}%2Bcu{cuda_short}torch{torch_mm}-{py_tag}-{py_tag}-win_amd64.whl"
-        platforms: [win_amd64]
+
+  gsplat:
+    method: index
+    index_url: "https://pozzettiandrea.github.io/cuda-wheels"
+    description: Gaussian splatting rasterization
+
+  cc_torch:
+    method: index
+    index_url: "https://pozzettiandrea.github.io/cuda-wheels"
+    description: GPU-accelerated connected components
+
+  torch_generic_nms:
+    method: index
+    index_url: "https://pozzettiandrea.github.io/cuda-wheels"
+    description: GPU-accelerated Non-Maximum Suppression
 
   # ===========================================================================
   # spconv - PyPI with CUDA-versioned package names
@@ -103,15 +113,9 @@ packages:
   # sageattention - Fast quantized attention
   # ===========================================================================
   sageattention:
-    method: github_release
+    method: index
+    index_url: "https://pozzettiandrea.github.io/cuda-wheels"
     description: SageAttention - 2-5x faster than FlashAttention with quantized kernels
-    sources:
-      - name: kijai-hf
-        url_template: "https://huggingface.co/Kijai/PrecompiledWheels/resolve/main/sageattention-{version}-cp312-cp312-linux_x86_64.whl"
-        platforms: [linux_x86_64]
-      - name: woct0rdho
-        url_template: "https://github.com/woct0rdho/SageAttention/releases/download/v2.2.0-windows.post3/sageattention-2.2.0%2Bcu{cuda_short}torch{torch_version}.post3-cp39-abi3-win_amd64.whl"
-        platforms: [win_amd64]
 
   # ===========================================================================
   # triton - Required for sageattention on Linux