comfy-env 0.0.25__py3-none-any.whl → 0.0.27__py3-none-any.whl

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/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

{comfy_env-0.0.25.dist-info → comfy_env-0.0.27.dist-info}/METADATA

@@ -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.dist-info → comfy_env-0.0.27.dist-info}/RECORD

@@ -2,7 +2,7 @@ comfy_env/__init__.py,sha256=u2uTyoysPQyNMcRp5U4VTMJF11FBW6Goqu0DN-BdUuY,3678
 comfy_env/cli.py,sha256=X-GCQMP0mtMcE3ZgkT-VLQ4Gq3UUvcb_Ux_NClEFhgI,15975
 comfy_env/decorator.py,sha256=6JCKwLHaZtOLVDexs_gh_-NtS2ZK0V7nGCPqkyeYEAA,16688
 comfy_env/errors.py,sha256=8hN8NDlo8oBUdapc-eT3ZluigI5VBzfqsSBvQdfWlz4,9943
-comfy_env/install.py,sha256=pyyaYbCFZn6T5_sgjumd1TeFaTEXUTcdFbex31m_JEc,25837
+comfy_env/install.py,sha256=VuIPPqcERIi_ZNHfQqduKUt5bKhsnKxymkmEZ2YG0Hk,28008
 comfy_env/nodes.py,sha256=CWUe35jU5SKk4ur-SddZePdqWgxJDlxGhpcJiu5pAK4,4354
 comfy_env/pixi.py,sha256=Oj9Z9bBg8IpsPF3usAx9v4w6OhpNk4LQjUuAP5ohnQY,13324
 comfy_env/registry.py,sha256=uFCtGmWYvwGCqObXgzmArX7o5JsFNsHXxayofk3m6no,2569
@@ -33,9 +33,9 @@ comfy_env/workers/pool.py,sha256=MtjeOWfvHSCockq8j1gfnxIl-t01GSB79T5N4YB82Lg,695
 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=_ekHfZPqBIPY08DjqiXm6cTBQH4DrbxRWR3AAv3mit8,31589
-comfy_env/wheel_sources.yml,sha256=nSZ8XB_I5JXQGB7AgC6lHs_IXMd9Kcno10artNL8BKw,7775
-comfy_env-0.0.25.dist-info/METADATA,sha256=IcYAcYC42bhuCU4ijkeY1CfUwX4b3AVRhnFZ3mR1BXw,5400
-comfy_env-0.0.25.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-comfy_env-0.0.25.dist-info/entry_points.txt,sha256=J4fXeqgxU_YenuW_Zxn_pEL7J-3R0--b6MS5t0QmAr0,49
-comfy_env-0.0.25.dist-info/licenses/LICENSE,sha256=E68QZMMpW4P2YKstTZ3QU54HRQO8ecew09XZ4_Vn870,1093
-comfy_env-0.0.25.dist-info/RECORD,,
+comfy_env/wheel_sources.yml,sha256=O9DJhhZnaTbV9A3Kidq8T4VntxyGAod0epT2dK-43N8,7139
+comfy_env-0.0.27.dist-info/METADATA,sha256=xvvoXEbukww7s5g7dRCYtxOf72QDnPGQvm7xohLiV5E,5400
+comfy_env-0.0.27.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+comfy_env-0.0.27.dist-info/entry_points.txt,sha256=J4fXeqgxU_YenuW_Zxn_pEL7J-3R0--b6MS5t0QmAr0,49
+comfy_env-0.0.27.dist-info/licenses/LICENSE,sha256=E68QZMMpW4P2YKstTZ3QU54HRQO8ecew09XZ4_Vn870,1093
+comfy_env-0.0.27.dist-info/RECORD,,