comfy-env 0.0.8__py3-none-any.whl → 0.0.16__py3-none-any.whl

comfy_env/registry.py

@@ -1,8 +1,7 @@
 """Built-in registry of CUDA packages and their wheel sources.
 
-This module provides a mapping of well-known CUDA packages to their
-installation sources, eliminating the need for users to specify
-wheel_sources in their comfyui_env.toml.
+This module loads package configurations from wheel_sources.yml and provides
+lookup functions for the install module.
 
 Install method types:
 - "index": Use pip --extra-index-url (PEP 503 simple repository)
@@ -10,9 +9,13 @@ Install method types:
 - "find_links": Use pip --find-links (for PyG, etc.)
 - "pypi_variant": Package name varies by CUDA version (e.g., spconv-cu124)
 - "github_release": Direct wheel URL from GitHub releases with fallback sources
+- "pypi": Standard PyPI install
 """
 
-from typing import Dict, Any, Optional
+from pathlib import Path
+from typing import Any, Dict, Optional
+
+import yaml
 
 
 def get_cuda_short2(cuda_version: str) -> str:
@@ -40,180 +43,16 @@ def get_cuda_short2(cuda_version: str) -> str:
     return f"{major}{minor}"
 
 
-# =============================================================================
-# Package Registry
-# =============================================================================
-# Maps package names to their installation configuration.
-#
-# Template variables available:
-#   {cuda_version}  - Full CUDA version (e.g., "12.8")
-#   {cuda_short}    - CUDA without dot (e.g., "128")
-#   {cuda_short2}   - CUDA short for spconv (e.g., "124" not "1240")
-#   {torch_version} - Full PyTorch version (e.g., "2.8.0")
-#   {torch_short}   - PyTorch without dots (e.g., "280")
-#   {torch_mm}      - PyTorch major.minor (e.g., "28")
-#   {py_version}    - Python version (e.g., "3.10")
-#   {py_short}      - Python without dot (e.g., "310")
-#   {py_minor}      - Python minor version only (e.g., "10")
-#   {platform}      - Platform tag (e.g., "linux_x86_64")
-# =============================================================================
-
-PACKAGE_REGISTRY: Dict[str, Dict[str, Any]] = {
-    # =========================================================================
-    # PyTorch Geometric (PyG) packages - official index
-    # https://pytorch-geometric.readthedocs.io/en/latest/install/installation.html
-    # Uses --find-links (not --extra-index-url) for proper wheel discovery
-    # =========================================================================
-    "torch-scatter": {
-        "method": "find_links",
-        "index_url": "https://data.pyg.org/whl/torch-{torch_version}+cu{cuda_short}.html",
-        "description": "Scatter operations for PyTorch",
-    },
-    "torch-cluster": {
-        "method": "find_links",
-        "index_url": "https://data.pyg.org/whl/torch-{torch_version}+cu{cuda_short}.html",
-        "description": "Clustering algorithms for PyTorch",
-    },
-    "torch-sparse": {
-        "method": "find_links",
-        "index_url": "https://data.pyg.org/whl/torch-{torch_version}+cu{cuda_short}.html",
-        "description": "Sparse tensor operations for PyTorch",
-    },
-    "torch-spline-conv": {
-        "method": "find_links",
-        "index_url": "https://data.pyg.org/whl/torch-{torch_version}+cu{cuda_short}.html",
-        "description": "Spline convolutions for PyTorch",
-    },
-
-    # =========================================================================
-    # pytorch3d - Facebook's official wheels
-    # https://github.com/facebookresearch/pytorch3d/blob/main/INSTALL.md
-    # =========================================================================
-    "pytorch3d": {
-        "method": "index",
-        "index_url": "https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/py3{py_minor}_cu{cuda_short}_pyt{torch_short}/download.html",
-        "description": "PyTorch3D - 3D deep learning library",
-    },
-
-    # =========================================================================
-    # PozzettiAndrea wheel repos (GitHub Pages indexes)
-    # =========================================================================
-    # nvdiffrast - wheels are now at cu{cuda}-torch{torch_short} releases
-    "nvdiffrast": {
-        "method": "github_index",
-        "index_url": "https://pozzettiandrea.github.io/nvdiffrast-full-wheels/cu{cuda_short}-torch{torch_short}/",
-        "description": "NVIDIA differentiable rasterizer",
-    },
-    # cumesh, o_voxel, flex_gemm, nvdiffrec_render use torch_short (3 digits: 280)
-    "cumesh": {
-        "method": "github_index",
-        "index_url": "https://pozzettiandrea.github.io/cumesh-wheels/cu{cuda_short}-torch{torch_short}/",
-        "description": "CUDA-accelerated mesh utilities",
-    },
-    "o_voxel": {
-        "method": "github_index",
-        "index_url": "https://pozzettiandrea.github.io/ovoxel-wheels/cu{cuda_short}-torch{torch_short}/",
-        "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}/",
-        "description": "Flexible GEMM operations",
-    },
-    "nvdiffrec_render": {
-        "method": "github_release",
-        "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", "windows_amd64"],
-            },
-        ],
-        "description": "NVDiffRec rendering utilities",
-    },
-
-    # =========================================================================
-    # spconv - PyPI with CUDA-versioned package names
-    # Package names: spconv-cu118, spconv-cu120, spconv-cu121, spconv-cu124, spconv-cu126
-    # Note: Max available is cu126 as of Jan 2026, use explicit spconv-cu126 in config
-    # =========================================================================
-    "spconv": {
-        "method": "pypi_variant",
-        "package_template": "spconv-cu{cuda_short2}",
-        "description": "Sparse convolution library (use spconv-cu126 for CUDA 12.6+)",
-    },
-
-    # =========================================================================
-    # sageattention - Fast quantized attention (2-5x faster than FlashAttention)
-    # Linux: Prebuilt wheels from Kijai/PrecompiledWheels (v2.2.0, cp312)
-    # Windows: Prebuilt wheels from woct0rdho (v2.2.0, cp39-abi3)
-    # =========================================================================
-    "sageattention": {
-        "method": "github_release",
-        "sources": [
-            # Linux: Kijai's precompiled wheels on HuggingFace (Python 3.12)
-            {
-                "name": "kijai-hf",
-                "url_template": "https://huggingface.co/Kijai/PrecompiledWheels/resolve/main/sageattention-{version}-cp312-cp312-linux_x86_64.whl",
-                "platforms": ["linux_x86_64"],
-            },
-            # Windows: woct0rdho prebuilt wheels (ABI3: Python >= 3.9)
-            # Format: sageattention-2.2.0+cu128torch2.8.0.post3-cp39-abi3-win_amd64.whl
-            {
-                "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"],
-            },
-        ],
-        "description": "SageAttention - 2-5x faster than FlashAttention with quantized kernels",
-    },
-
-    # =========================================================================
-    # triton - Required for sageattention on Linux (usually bundled with PyTorch)
-    # =========================================================================
-    "triton": {
-        "method": "pypi",
-        "description": "Triton compiler for custom CUDA kernels (required by sageattention)",
-    },
-
-    # =========================================================================
-    # flash-attn - Multi-source prebuilt wheels
-    # Required for UniRig and other transformer-based models
-    # Sources: Dao-AILab (official), mjun0812 (Linux), bdashore3 (Windows)
-    # =========================================================================
-    "flash-attn": {
-        "method": "github_release",
-        "sources": [
-            # Linux: Dao-AILab official wheels (CUDA 12.x, PyTorch 2.4-2.8)
-            # Format: flash_attn-2.8.3+cu12torch2.8cxx11abiTRUE-cp310-cp310-linux_x86_64.whl
-            {
-                "name": "Dao-AILab",
-                "url_template": "https://github.com/Dao-AILab/flash-attention/releases/download/v{version}/flash_attn-{version}%2Bcu{cuda_major}torch{torch_dotted_mm}cxx11abiTRUE-{py_tag}-{py_tag}-linux_x86_64.whl",
-                "platforms": ["linux_x86_64"],
-            },
-            # Linux: mjun0812 prebuilt wheels (CUDA 12.4-13.0, PyTorch 2.5-2.9)
-            # Format: flash_attn-2.8.3+cu128torch2.8-cp310-cp310-linux_x86_64.whl
-            # Note: Release v0.7.2 contains multiple flash_attn versions
-            {
-                "name": "mjun0812",
-                "url_template": "https://github.com/mjun0812/flash-attention-prebuild-wheels/releases/download/v0.7.2/flash_attn-{version}%2Bcu{cuda_short}torch{torch_dotted_mm}-{py_tag}-{py_tag}-linux_x86_64.whl",
-                "platforms": ["linux_x86_64"],
-            },
-            # Windows: bdashore3 prebuilt wheels (CUDA 12.4/12.8, PyTorch 2.6-2.8)
-            {
-                "name": "bdashore3",
-                "url_template": "https://github.com/bdashore3/flash-attention/releases/download/v{version}/flash_attn-{version}%2Bcu{cuda_short}torch{torch_version}cxx11abiFALSE-{py_tag}-{py_tag}-win_amd64.whl",
-                "platforms": ["win_amd64"],
-            },
-        ],
-        "description": "Flash Attention for fast transformer inference",
-    },
-}
+def _load_wheel_sources() -> Dict[str, Dict[str, Any]]:
+    """Load package registry from wheel_sources.yml."""
+    yml_path = Path(__file__).parent / "wheel_sources.yml"
+    with open(yml_path, "r") as f:
+        data = yaml.safe_load(f)
+    return data.get("packages", {})
+
+
+# Load registry at module import time
+PACKAGE_REGISTRY: Dict[str, Dict[str, Any]] = _load_wheel_sources()
 
 
 def get_package_info(package: str) -> Optional[Dict[str, Any]]:

comfy_env/resolver.py

@@ -26,7 +26,7 @@ from pathlib import Path
 from typing import Dict, List, Optional, Tuple
 from urllib.parse import urlparse
 
-from .env.detection import detect_cuda_version, detect_gpu_info
+from .env.cuda_gpu_detection import detect_cuda_version, detect_gpu_info
 
 
 @dataclass
@@ -137,11 +137,14 @@ class RuntimeEnv:
             "py_version": self.python_version,
             "py_short": self.python_short,
             "py_minor": py_minor,
+            "py_tag": f"cp{self.python_short}",  # e.g., cp310, cp311
         }
 
         if self.cuda_version:
             result["cuda_version"] = self.cuda_version
             result["cuda_short"] = self.cuda_short
+            # cuda_major: just the major version (e.g., "12" from "12.8")
+            result["cuda_major"] = self.cuda_version.split(".")[0]
 
         if self.torch_version:
             result["torch_version"] = self.torch_version
@@ -218,21 +221,6 @@ class WheelSource:
         return package.lower() in [p.lower() for p in self.packages]
 
 
-# Default wheel sources for common CUDA packages
-DEFAULT_WHEEL_SOURCES = [
-    WheelSource(
-        name="nvdiffrast-wheels",
-        url_template="https://github.com/PozzettiAndrea/nvdiffrast-full-wheels/releases/download/v{version}/nvdiffrast-{version}%2Bcu{cuda_short}torch{torch_mm}-cp{py_short}-cp{py_short}-{platform}.whl",
-        packages=["nvdiffrast"],
-    ),
-    WheelSource(
-        name="cumesh-wheels",
-        url_template="https://github.com/PozzettiAndrea/cumesh-wheels/releases/download/v{version}/{package}-{version}%2Bcu{cuda_short}torch{torch_mm}-cp{py_short}-cp{py_short}-{platform}.whl",
-        packages=["pytorch3d", "torch-cluster", "torch-scatter", "torch-sparse"],
-    ),
-]
-
-
 class WheelResolver:
     """
     Resolves CUDA wheel URLs from package name and runtime environment.
@@ -255,10 +243,11 @@ class WheelResolver:
         Initialize resolver.
 
         Args:
-            sources: List of WheelSource configurations.
+            sources: List of WheelSource configurations. Defaults to empty
+                    (use PACKAGE_REGISTRY in install.py for actual sources).
             overrides: Package-specific URL overrides (package -> template).
         """
-        self.sources = sources or DEFAULT_WHEEL_SOURCES
+        self.sources = sources or []
         self.overrides = overrides or {}
 
     def resolve(

comfy_env/stubs/folder_paths.py

@@ -52,6 +52,10 @@ class _ModuleProxy:
         base = _find_comfyui_base()
         return str(base / "input") if base else None
 
+    def get_temp_directory(self):
+        base = _find_comfyui_base()
+        return str(base / "temp") if base else None
+
 # Replace module with proxy instance
 import sys
 sys.modules[__name__] = _ModuleProxy()

comfy_env/tools.py

@@ -0,0 +1,221 @@
+"""
+Tool installers for external dependencies like Blender.
+
+Usage in comfy-env.toml:
+    [tools]
+    blender = "4.2"
+"""
+
+import os
+import platform
+import shutil
+import subprocess
+import tarfile
+import zipfile
+from pathlib import Path
+from typing import Callable, Optional
+from urllib.request import urlretrieve
+
+from .env.config import ToolConfig
+
+# Default install location
+DEFAULT_TOOLS_DIR = Path.home() / ".comfy-env" / "tools"
+
+# Blender download URLs by platform and version
+BLENDER_DOWNLOADS = {
+    "4.2": {
+        "linux": "https://download.blender.org/release/Blender4.2/blender-4.2.0-linux-x64.tar.xz",
+        "windows": "https://download.blender.org/release/Blender4.2/blender-4.2.0-windows-x64.zip",
+        "darwin": "https://download.blender.org/release/Blender4.2/blender-4.2.0-macos-arm64.dmg",
+    },
+    "4.3": {
+        "linux": "https://download.blender.org/release/Blender4.3/blender-4.3.0-linux-x64.tar.xz",
+        "windows": "https://download.blender.org/release/Blender4.3/blender-4.3.0-windows-x64.zip",
+        "darwin": "https://download.blender.org/release/Blender4.3/blender-4.3.0-macos-arm64.dmg",
+    },
+}
+
+
+def get_platform() -> str:
+    """Get current platform name."""
+    system = platform.system().lower()
+    if system == "linux":
+        return "linux"
+    elif system == "windows":
+        return "windows"
+    elif system == "darwin":
+        return "darwin"
+    return system
+
+
+def install_tool(
+    config: ToolConfig,
+    log: Callable[[str], None] = print,
+    base_dir: Optional[Path] = None,
+) -> Optional[Path]:
+    """Install a tool based on its config.
+
+    Args:
+        config: Tool configuration
+        log: Logging callback
+        base_dir: Base directory for tools. Tools install to base_dir/tools/<name>/
+    """
+    if config.name.lower() == "blender":
+        return install_blender(config.version, log, config.install_dir or base_dir)
+    else:
+        log(f"Unknown tool: {config.name}")
+        return None
+
+
+def install_blender(
+    version: str = "4.2",
+    log: Callable[[str], None] = print,
+    base_dir: Optional[Path] = None,
+) -> Optional[Path]:
+    """
+    Install Blender to the specified directory.
+
+    Args:
+        version: Blender version to install (e.g., "4.2")
+        log: Logging callback
+        base_dir: Base directory. Blender installs to base_dir/tools/blender/
+                  If None, uses ~/.comfy-env/tools/blender/
+
+    Returns path to blender executable if successful.
+    """
+    plat = get_platform()
+    if base_dir:
+        install_dir = base_dir / "tools" / "blender"
+    else:
+        install_dir = DEFAULT_TOOLS_DIR / "blender"
+
+    # Check if already installed
+    exe = find_blender(install_dir)
+    if exe:
+        log(f"Blender already installed: {exe}")
+        return exe
+
+    # Get download URL
+    if version not in BLENDER_DOWNLOADS:
+        log(f"Unknown Blender version: {version}. Available: {list(BLENDER_DOWNLOADS.keys())}")
+        return None
+
+    urls = BLENDER_DOWNLOADS[version]
+    if plat not in urls:
+        log(f"Blender {version} not available for {plat}")
+        return None
+
+    url = urls[plat]
+    log(f"Downloading Blender {version} for {plat}...")
+
+    install_dir.mkdir(parents=True, exist_ok=True)
+    archive_name = url.split("/")[-1]
+    archive_path = install_dir / archive_name
+
+    try:
+        # Download
+        urlretrieve(url, archive_path)
+        log(f"Downloaded to {archive_path}")
+
+        # Extract
+        log("Extracting...")
+        if archive_name.endswith(".tar.xz"):
+            with tarfile.open(archive_path, "r:xz") as tar:
+                tar.extractall(install_dir)
+        elif archive_name.endswith(".zip"):
+            with zipfile.ZipFile(archive_path, "r") as zf:
+                zf.extractall(install_dir)
+        elif archive_name.endswith(".dmg"):
+            # macOS DMG requires special handling
+            log("macOS DMG installation not yet automated. Please install manually.")
+            archive_path.unlink()
+            return None
+
+        # Clean up archive
+        archive_path.unlink()
+
+        # Find the executable
+        exe = find_blender(install_dir)
+        if exe:
+            log(f"Blender installed: {exe}")
+            return exe
+        else:
+            log("Blender extracted but executable not found")
+            return None
+
+    except Exception as e:
+        log(f"Failed to install Blender: {e}")
+        if archive_path.exists():
+            archive_path.unlink()
+        return None
+
+
+def find_blender(search_dir: Optional[Path] = None) -> Optional[Path]:
+    """Find Blender executable."""
+    # Check PATH first
+    blender_in_path = shutil.which("blender")
+    if blender_in_path:
+        return Path(blender_in_path)
+
+    # Check common locations
+    plat = get_platform()
+    search_paths = []
+
+    if search_dir:
+        search_paths.append(search_dir)
+
+    search_paths.append(DEFAULT_TOOLS_DIR / "blender")
+
+    if plat == "windows":
+        search_paths.extend([
+            Path(os.environ.get("ProgramFiles", "C:/Program Files")) / "Blender Foundation",
+            Path.home() / "AppData" / "Local" / "Blender Foundation",
+        ])
+    elif plat == "linux":
+        search_paths.extend([
+            Path("/opt/blender"),
+            Path.home() / "blender",
+        ])
+    elif plat == "darwin":
+        search_paths.append(Path("/Applications/Blender.app/Contents/MacOS"))
+
+    exe_name = "blender.exe" if plat == "windows" else "blender"
+
+    for base in search_paths:
+        if not base.exists():
+            continue
+        # Direct check
+        exe = base / exe_name
+        if exe.exists():
+            return exe
+        # Search subdirectories (for extracted archives like blender-4.2.0-linux-x64/)
+        for subdir in base.iterdir():
+            if subdir.is_dir():
+                exe = subdir / exe_name
+                if exe.exists():
+                    return exe
+
+    return None
+
+
+def ensure_blender(
+    version: str = "4.2",
+    log: Callable[[str], None] = print,
+    base_dir: Optional[Path] = None,
+) -> Optional[Path]:
+    """Ensure Blender is installed, installing if necessary.
+
+    Args:
+        version: Blender version to install
+        log: Logging callback
+        base_dir: Base directory. Searches/installs in base_dir/tools/blender/
+    """
+    if base_dir:
+        search_dir = base_dir / "tools" / "blender"
+    else:
+        search_dir = DEFAULT_TOOLS_DIR / "blender"
+
+    exe = find_blender(search_dir)
+    if exe:
+        return exe
+    return install_blender(version, log, base_dir)

comfy_env/wheel_sources.yml

@@ -0,0 +1,141 @@
+# Wheel sources registry for CUDA packages
+#
+# Template variables:
+#   {cuda_version}   - Full CUDA version (e.g., "12.8")
+#   {cuda_short}     - CUDA without dot (e.g., "128")
+#   {cuda_short2}    - CUDA short for spconv (e.g., "124" not "1240")
+#   {cuda_major}     - CUDA major version (e.g., "12")
+#   {torch_version}  - Full PyTorch version (e.g., "2.8.0")
+#   {torch_short}    - PyTorch without dots (e.g., "280")
+#   {torch_mm}       - PyTorch major.minor no dot (e.g., "28")
+#   {torch_dotted_mm}- PyTorch major.minor with dot (e.g., "2.8")
+#   {py_version}     - Python version (e.g., "3.10")
+#   {py_short}       - Python without dot (e.g., "310")
+#   {py_minor}       - Python minor version only (e.g., "10")
+#   {py_tag}         - Python tag (e.g., "cp310")
+#   {platform}       - Platform tag (e.g., "linux_x86_64")
+#
+# Install methods:
+#   index          - pip --extra-index-url (PEP 503)
+#   find_links     - pip --find-links
+#   github_index   - GitHub Pages index (--find-links)
+#   pypi_variant   - Package name varies by CUDA (e.g., spconv-cu124)
+#   github_release - Direct wheel URL from GitHub releases
+#   pypi           - Standard PyPI install
+
+packages:
+  # ===========================================================================
+  # PyTorch Geometric (PyG) - official index
+  # https://pytorch-geometric.readthedocs.io/en/latest/install/installation.html
+  # ===========================================================================
+  torch-scatter:
+    method: find_links
+    index_url: "https://data.pyg.org/whl/torch-{torch_version}+cu{cuda_short}.html"
+    description: Scatter operations for PyTorch
+
+  torch-cluster:
+    method: find_links
+    index_url: "https://data.pyg.org/whl/torch-{torch_version}+cu{cuda_short}.html"
+    description: Clustering algorithms for PyTorch
+
+  torch-sparse:
+    method: find_links
+    index_url: "https://data.pyg.org/whl/torch-{torch_version}+cu{cuda_short}.html"
+    description: Sparse tensor operations for PyTorch
+
+  torch-spline-conv:
+    method: find_links
+    index_url: "https://data.pyg.org/whl/torch-{torch_version}+cu{cuda_short}.html"
+    description: Spline convolutions for PyTorch
+
+  # ===========================================================================
+  # pytorch3d - Facebook's official wheels
+  # https://github.com/facebookresearch/pytorch3d/blob/main/INSTALL.md
+  # ===========================================================================
+  pytorch3d:
+    method: index
+    index_url: "https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/py3{py_minor}_cu{cuda_short}_pyt{torch_short}/download.html"
+    description: PyTorch3D - 3D deep learning library
+
+  # ===========================================================================
+  # PozzettiAndrea wheel repos (GitHub Pages indexes)
+  # ===========================================================================
+  nvdiffrast:
+    method: github_index
+    index_url: "https://pozzettiandrea.github.io/nvdiffrast-full-wheels/cu{cuda_short}-torch{torch_short}/"
+    description: NVIDIA differentiable rasterizer
+
+  cumesh:
+    method: github_index
+    index_url: "https://pozzettiandrea.github.io/cumesh-wheels/cu{cuda_short}-torch{torch_short}/"
+    description: CUDA-accelerated mesh utilities
+
+  o_voxel:
+    method: github_index
+    index_url: "https://pozzettiandrea.github.io/ovoxel-wheels/cu{cuda_short}-torch{torch_short}/"
+    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}/"
+    description: Flexible GEMM operations
+
+  nvdiffrec_render:
+    method: github_release
+    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]
+
+  # ===========================================================================
+  # spconv - PyPI with CUDA-versioned package names
+  # ===========================================================================
+  spconv:
+    method: pypi_variant
+    package_template: "spconv-cu{cuda_short2}"
+    description: Sparse convolution library (use spconv-cu126 for CUDA 12.6+)
+
+  # ===========================================================================
+  # sageattention - Fast quantized attention
+  # ===========================================================================
+  sageattention:
+    method: github_release
+    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
+  # ===========================================================================
+  triton:
+    method: pypi
+    description: Triton compiler for custom CUDA kernels
+
+  # ===========================================================================
+  # flash-attn - Multi-source prebuilt wheels
+  # ===========================================================================
+  flash-attn:
+    method: github_release
+    description: Flash Attention for fast transformer inference
+    sources:
+      - name: Dao-AILab
+        url_template: "https://github.com/Dao-AILab/flash-attention/releases/download/v{version}/flash_attn-{version}%2Bcu{cuda_major}torch{torch_dotted_mm}cxx11abiTRUE-{py_tag}-{py_tag}-linux_x86_64.whl"
+        platforms: [linux_x86_64]
+      - name: mjun0812
+        url_template: "https://github.com/mjun0812/flash-attention-prebuild-wheels/releases/download/v0.7.7/flash_attn-{version}%2Bcu{cuda_short}torch{torch_dotted_mm}-{py_tag}-{py_tag}-linux_x86_64.whl"
+        platforms: [linux_x86_64]
+      - name: mjun0812-windows
+        url_template: "https://github.com/mjun0812/flash-attention-prebuild-wheels/releases/download/v0.7.7/flash_attn-{version}%2Bcu{cuda_short}torch{torch_dotted_mm}-{py_tag}-{py_tag}-win_amd64.whl"
+        platforms: [win_amd64]
+      - name: bdashore3
+        url_template: "https://github.com/bdashore3/flash-attention/releases/download/v{version}/flash_attn-{version}%2Bcu{cuda_short}torch{torch_version}cxx11abiFALSE-{py_tag}-{py_tag}-win_amd64.whl"
+        platforms: [win_amd64]

{comfy_env-0.0.8.dist-info → comfy_env-0.0.16.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: comfy-env
-Version: 0.0.8
+Version: 0.0.16
 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
@@ -17,6 +17,7 @@ Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
 Requires-Python: >=3.10
+Requires-Dist: pyyaml>=6.0
 Requires-Dist: tomli>=2.0.0; python_version < '3.11'
 Requires-Dist: uv>=0.4.0
 Provides-Extra: dev