comfy-env 0.0.41__py3-none-any.whl → 0.0.43__py3-none-any.whl

comfy_env/registry.py

@@ -3,13 +3,9 @@
 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)
-- "github_index": GitHub Pages index (--find-links)
-- "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
+Each package has either:
+- wheel_template: Direct URL template for .whl file
+- package_name: PyPI package name template (for packages like spconv-cu124)
 """
 
 from pathlib import Path
@@ -34,8 +30,6 @@ def get_cuda_short2(cuda_version: str) -> str:
         '124'
         >>> get_cuda_short2("12.8")
         '128'
-        >>> get_cuda_short2("11.8")
-        '118'
     """
     parts = cuda_version.split(".")
     major = parts[0]
@@ -89,3 +83,48 @@ def is_registered(package: str) -> bool:
         True if package is registered
     """
     return package.lower() in PACKAGE_REGISTRY
+
+
+def get_wheel_template(package: str) -> Optional[str]:
+    """Get wheel_template for a package.
+
+    Args:
+        package: Package name (case-insensitive)
+
+    Returns:
+        wheel_template string or None if not found/not available
+    """
+    info = get_package_info(package)
+    if info:
+        return info.get("wheel_template")
+    return None
+
+
+def get_package_name_template(package: str) -> Optional[str]:
+    """Get package_name template for PyPI variant packages (like spconv).
+
+    Args:
+        package: Package name (case-insensitive)
+
+    Returns:
+        package_name template string or None if not found/not available
+    """
+    info = get_package_info(package)
+    if info:
+        return info.get("package_name")
+    return None
+
+
+def get_default_version(package: str) -> Optional[str]:
+    """Get default_version for a package.
+
+    Args:
+        package: Package name (case-insensitive)
+
+    Returns:
+        default_version string or None if not specified
+    """
+    info = get_package_info(package)
+    if info:
+        return info.get("default_version")
+    return None

comfy_env/resolver.py

@@ -1,30 +1,20 @@
 """
-Wheel URL resolver for CUDA-compiled packages.
+Runtime environment detection for wheel resolution.
 
-This module provides deterministic wheel URL construction based on the runtime
-environment (CUDA version, PyTorch version, Python version, platform).
-
-Unlike pip's constraint solver, this module constructs exact URLs from templates
-and validates that they exist. If a wheel doesn't exist, it fails fast with
-a clear error message.
+This module provides RuntimeEnv for detecting the current system environment
+(CUDA version, PyTorch version, Python version, platform).
 
 Example:
-    from comfy_env.resolver import WheelResolver, RuntimeEnv
+    from comfy_env.resolver import RuntimeEnv
 
     env = RuntimeEnv.detect()
-    resolver = WheelResolver()
-
-    url = resolver.resolve("nvdiffrast", version="0.4.0", env=env)
-    # Returns: https://github.com/.../nvdiffrast-0.4.0+cu128torch28-cp310-...-linux_x86_64.whl
+    print(f"CUDA: {env.cuda_version}, PyTorch: {env.torch_version}")
 """
 
 import platform
-import re
 import sys
-from dataclasses import dataclass, field
-from pathlib import Path
-from typing import Dict, List, Optional, Tuple
-from urllib.parse import urlparse
+from dataclasses import dataclass
+from typing import Dict, Optional, Tuple
 
 from .env.cuda_gpu_detection import detect_cuda_version, detect_gpu_info
 
@@ -106,8 +96,8 @@ class RuntimeEnv:
         try:
             gpu_info = detect_gpu_info()
             if gpu_info:
-                gpu_name = gpu_info.get("name")
-                gpu_compute = gpu_info.get("compute_capability")
+                gpu_name = gpu_info[0].get("name") if isinstance(gpu_info, list) else gpu_info.get("name")
+                gpu_compute = gpu_info[0].get("compute_capability") if isinstance(gpu_info, list) else gpu_info.get("compute_capability")
         except Exception:
             pass
 
@@ -127,7 +117,6 @@ class RuntimeEnv:
 
     def as_dict(self) -> Dict[str, str]:
         """Convert to dict for template substitution."""
-        # Extract py_minor from python_version (e.g., "3.10" -> "10")
         py_minor = self.python_version.split(".")[-1] if self.python_version else ""
 
         result = {
@@ -137,20 +126,18 @@ 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
+            "py_tag": f"cp{self.python_short}",
         }
 
         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
             result["torch_short"] = self.torch_short
             result["torch_mm"] = self.torch_mm
-            # torch_dotted_mm: "2.8" format (major.minor with dot) for flash-attn URLs
             parts = self.torch_version.split(".")[:2]
             result["torch_dotted_mm"] = ".".join(parts)
 
@@ -173,7 +160,6 @@ def _get_platform_tag() -> str:
     machine = platform.machine().lower()
 
     if sys.platform.startswith('linux'):
-        # Use manylinux tag
         if machine in ('x86_64', 'amd64'):
             return 'linux_x86_64'
         elif machine == 'aarch64':
@@ -186,7 +172,6 @@ def _get_platform_tag() -> str:
         return 'win32'
 
     elif sys.platform == 'darwin':
-        # macOS - use generic tag
         if machine == 'arm64':
             return 'macosx_11_0_arm64'
         return 'macosx_10_9_x86_64'
@@ -199,7 +184,6 @@ def _detect_torch_version() -> Optional[str]:
     try:
         import torch
         version = torch.__version__
-        # Strip CUDA suffix (e.g., "2.8.0+cu128" -> "2.8.0")
         if '+' in version:
             version = version.split('+')[0]
         return version
@@ -207,166 +191,6 @@ def _detect_torch_version() -> Optional[str]:
         return None
 
 
-@dataclass
-class WheelSource:
-    """Configuration for a wheel source (GitHub releases, custom index, etc.)."""
-    name: str
-    url_template: str
-    packages: List[str] = field(default_factory=list)  # Empty = all packages
-
-    def supports(self, package: str) -> bool:
-        """Check if this source provides the given package."""
-        if not self.packages:
-            return True  # Empty list = supports all
-        return package.lower() in [p.lower() for p in self.packages]
-
-
-class WheelResolver:
-    """
-    Resolves CUDA wheel URLs from package name and runtime environment.
-
-    Resolution strategy:
-    1. Check explicit overrides in config
-    2. Try configured wheel sources in order
-    3. Fail with actionable error message
-
-    This is NOT a constraint solver. It constructs deterministic URLs
-    based on exact version matches.
-    """
-
-    def __init__(
-        self,
-        sources: Optional[List[WheelSource]] = None,
-        overrides: Optional[Dict[str, str]] = None,
-    ):
-        """
-        Initialize resolver.
-
-        Args:
-            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 []
-        self.overrides = overrides or {}
-
-    def resolve(
-        self,
-        package: str,
-        version: str,
-        env: RuntimeEnv,
-        verify: bool = False,
-    ) -> str:
-        """
-        Resolve wheel URL for a package.
-
-        Args:
-            package: Package name (e.g., "nvdiffrast").
-            version: Package version (e.g., "0.4.0").
-            env: Runtime environment for template expansion.
-
-        Returns:
-            Fully resolved wheel URL.
-
-        Raises:
-            WheelNotFoundError: If no wheel URL could be constructed or verified.
-        """
-        from .errors import WheelNotFoundError
-
-        # Prepare template variables
-        variables = env.as_dict()
-        variables["package"] = package
-        variables["version"] = version
-
-        # 1. Check explicit override
-        if package.lower() in self.overrides:
-            url = self._substitute(self.overrides[package.lower()], variables)
-            if verify and not self._url_exists(url):
-                raise WheelNotFoundError(
-                    package=package,
-                    version=version,
-                    env=env,
-                    tried_urls=[url],
-                    reason="Override URL returned 404",
-                )
-            return url
-
-        # 2. Try wheel sources
-        tried_urls = []
-        for source in self.sources:
-            if not source.supports(package):
-                continue
-
-            url = self._substitute(source.url_template, variables)
-            tried_urls.append(url)
-
-            if verify:
-                if self._url_exists(url):
-                    return url
-            else:
-                return url
-
-        # 3. Fail with helpful error
-        raise WheelNotFoundError(
-            package=package,
-            version=version,
-            env=env,
-            tried_urls=tried_urls,
-            reason="No wheel source found for package",
-        )
-
-    def resolve_all(
-        self,
-        packages: Dict[str, str],
-        env: RuntimeEnv,
-        verify: bool = False,
-    ) -> Dict[str, str]:
-        """
-        Resolve URLs for multiple packages.
-
-        Args:
-            packages: Dict of package -> version.
-            env: Runtime environment.
-            verify: Whether to verify URLs exist.
-
-        Returns:
-            Dict of package -> resolved URL.
-
-        Raises:
-            WheelNotFoundError: If any package cannot be resolved.
-        """
-        results = {}
-        for package, version in packages.items():
-            results[package] = self.resolve(package, version, env, verify=verify)
-        return results
-
-    def _substitute(self, template: str, variables: Dict[str, str]) -> str:
-        """
-        Substitute variables into URL template.
-
-        Handles both {var} and {var_name} style placeholders.
-        Missing variables are left as-is (caller should validate).
-        """
-        result = template
-        for key, value in variables.items():
-            result = result.replace(f"{{{key}}}", str(value))
-        return result
-
-    def _url_exists(self, url: str, timeout: float = 10.0) -> bool:
-        """
-        Check if a URL exists using HTTP HEAD request.
-
-        Returns True if URL returns 200 OK.
-        """
-        try:
-            import urllib.request
-            request = urllib.request.Request(url, method='HEAD')
-            with urllib.request.urlopen(request, timeout=timeout) as response:
-                return response.status == 200
-        except Exception:
-            return False
-
-
 def parse_wheel_requirement(req: str) -> Tuple[str, Optional[str]]:
     """
     Parse a wheel requirement string.
@@ -379,7 +203,6 @@ def parse_wheel_requirement(req: str) -> Tuple[str, Optional[str]]:
     Returns:
         Tuple of (package_name, version_or_None).
     """
-    # Handle version specifiers
     for op in ['==', '>=', '<=', '~=', '!=', '>', '<']:
         if op in req:
             parts = req.split(op, 1)

comfy_env/wheel_sources.yml

@@ -1,9 +1,12 @@
 # Wheel sources registry for CUDA packages
 #
+# Each package has a single wheel_template that resolves to a direct .whl URL.
+# Users can override these in their comfy-env.toml [wheel_sources] section.
+#
 # Template variables:
+#   {version}        - Package version (e.g., "0.4.0")
 #   {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")
@@ -14,173 +17,119 @@
 #   {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
+  # PyTorch Geometric (PyG) - official wheels
   # 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"
+    wheel_template: "https://data.pyg.org/whl/torch-{torch_version}%2Bcu{cuda_short}/torch_scatter-{version}%2Bpt{torch_mm}cu{cuda_short}-{py_tag}-{py_tag}-{platform}.whl"
     description: Scatter operations for PyTorch
 
   torch-cluster:
-    method: find_links
-    index_url: "https://data.pyg.org/whl/torch-{torch_version}+cu{cuda_short}.html"
+    wheel_template: "https://data.pyg.org/whl/torch-{torch_version}%2Bcu{cuda_short}/torch_cluster-{version}%2Bpt{torch_mm}cu{cuda_short}-{py_tag}-{py_tag}-{platform}.whl"
     description: Clustering algorithms for PyTorch
 
   torch-sparse:
-    method: find_links
-    index_url: "https://data.pyg.org/whl/torch-{torch_version}+cu{cuda_short}.html"
+    wheel_template: "https://data.pyg.org/whl/torch-{torch_version}%2Bcu{cuda_short}/torch_sparse-{version}%2Bpt{torch_mm}cu{cuda_short}-{py_tag}-{py_tag}-{platform}.whl"
     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"
+    wheel_template: "https://data.pyg.org/whl/torch-{torch_version}%2Bcu{cuda_short}/torch_spline_conv-{version}%2Bpt{torch_mm}cu{cuda_short}-{py_tag}-{py_tag}-{platform}.whl"
     description: Spline convolutions for PyTorch
 
   # ===========================================================================
-  # pytorch3d - PyTorch3D 3D deep learning library
-  # https://pozzettiandrea.github.io/cuda-wheels
+  # PozzettiAndrea cuda-wheels (GitHub releases)
+  # https://github.com/PozzettiAndrea/cuda-wheels/releases
   # ===========================================================================
   pytorch3d:
-    method: index
-    index_url: "https://pozzettiandrea.github.io/cuda-wheels"
-    wheel_template: "https://pozzettiandrea.github.io/cuda-wheels/pytorch3d/pytorch3d-{version}%2Bcu{cuda_short}torch{torch_mm}-{py_tag}-{py_tag}-{platform}.whl"
+    wheel_template: "https://github.com/PozzettiAndrea/cuda-wheels/releases/download/pytorch3d-latest/pytorch3d-{version}%2Bcu{cuda_short}torch{torch_mm}-{py_tag}-{py_tag}-{platform}.whl"
     default_version: "0.7.9"
     description: PyTorch3D - 3D deep learning library
 
-  # ===========================================================================
-  # PozzettiAndrea cuda-wheels (unified PEP 503 index)
-  # https://pozzettiandrea.github.io/cuda-wheels
-  # Wheel naming: {package}-{version}+cu{cuda_short}torch{torch_mm}-{py_tag}-{py_tag}-{platform}.whl
-  # ===========================================================================
   nvdiffrast:
-    method: index
-    index_url: "https://pozzettiandrea.github.io/cuda-wheels"
-    wheel_template: "https://pozzettiandrea.github.io/cuda-wheels/nvdiffrast/nvdiffrast-{version}%2Bcu{cuda_short}torch{torch_mm}-{py_tag}-{py_tag}-{platform}.whl"
+    wheel_template: "https://github.com/PozzettiAndrea/cuda-wheels/releases/download/nvdiffrast-latest/nvdiffrast-{version}%2Bcu{cuda_short}torch{torch_mm}-{py_tag}-{py_tag}-{platform}.whl"
+    default_version: "0.4.0"
     description: NVIDIA differentiable rasterizer
 
   cumesh:
-    method: index
-    index_url: "https://pozzettiandrea.github.io/cuda-wheels"
-    wheel_template: "https://pozzettiandrea.github.io/cuda-wheels/cumesh/cumesh-{version}%2Bcu{cuda_short}torch{torch_mm}-{py_tag}-{py_tag}-{platform}.whl"
+    wheel_template: "https://github.com/PozzettiAndrea/cuda-wheels/releases/download/cumesh-latest/cumesh-{version}%2Bcu{cuda_short}torch{torch_mm}-{py_tag}-{py_tag}-{platform}.whl"
+    default_version: "0.0.1"
     description: CUDA-accelerated mesh utilities
 
   cubvh:
-    method: index
-    index_url: "https://pozzettiandrea.github.io/cuda-wheels"
+    wheel_template: "https://github.com/PozzettiAndrea/cuda-wheels/releases/download/cubvh-latest/cubvh-{version}%2Bcu{cuda_short}torch{torch_mm}-{py_tag}-{py_tag}-{platform}.whl"
+    default_version: "0.1.2"
     description: CUDA BVH acceleration for mesh operations and marching cubes
 
   o_voxel:
-    method: index
-    index_url: "https://pozzettiandrea.github.io/cuda-wheels"
-    wheel_template: "https://pozzettiandrea.github.io/cuda-wheels/o-voxel/o_voxel-{version}%2Bcu{cuda_short}torch{torch_mm}-{py_tag}-{py_tag}-{platform}.whl"
+    wheel_template: "https://github.com/PozzettiAndrea/cuda-wheels/releases/download/o_voxel-latest/o_voxel-{version}%2Bcu{cuda_short}torch{torch_mm}-{py_tag}-{py_tag}-{platform}.whl"
+    default_version: "0.0.1"
     description: O-Voxel CUDA extension for TRELLIS
 
   flex_gemm:
-    method: index
-    index_url: "https://pozzettiandrea.github.io/cuda-wheels"
-    wheel_template: "https://pozzettiandrea.github.io/cuda-wheels/flex-gemm/flex_gemm-{version}%2Bcu{cuda_short}torch{torch_mm}-{py_tag}-{py_tag}-{platform}.whl"
+    wheel_template: "https://github.com/PozzettiAndrea/cuda-wheels/releases/download/flex_gemm-latest/flex_gemm-{version}%2Bcu{cuda_short}torch{torch_mm}-{py_tag}-{py_tag}-{platform}.whl"
+    default_version: "1.0.0"
     description: Flexible GEMM operations
 
   nvdiffrec_render:
-    method: index
-    index_url: "https://pozzettiandrea.github.io/cuda-wheels"
-    wheel_template: "https://pozzettiandrea.github.io/cuda-wheels/nvdiffrec-render/nvdiffrec_render-{version}%2Bcu{cuda_short}torch{torch_mm}-{py_tag}-{py_tag}-{platform}.whl"
+    wheel_template: "https://github.com/PozzettiAndrea/cuda-wheels/releases/download/nvdiffrec_render-latest/nvdiffrec_render-{version}%2Bcu{cuda_short}torch{torch_mm}-{py_tag}-{py_tag}-{platform}.whl"
+    default_version: "0.0.1"
     description: NVDiffRec rendering utilities
 
   gsplat:
-    method: index
-    index_url: "https://pozzettiandrea.github.io/cuda-wheels"
-    wheel_template: "https://pozzettiandrea.github.io/cuda-wheels/gsplat/gsplat-{version}%2Bcu{cuda_short}torch{torch_mm}-{py_tag}-{py_tag}-{platform}.whl"
+    wheel_template: "https://github.com/PozzettiAndrea/cuda-wheels/releases/download/gsplat-latest/gsplat-{version}%2Bcu{cuda_short}torch{torch_mm}-{py_tag}-{py_tag}-{platform}.whl"
+    default_version: "1.5.3"
     description: Gaussian splatting rasterization
 
   cc_torch:
-    method: index
-    index_url: "https://pozzettiandrea.github.io/cuda-wheels"
-    wheel_template: "https://pozzettiandrea.github.io/cuda-wheels/cc-torch/cc_torch-{version}%2Bcu{cuda_short}torch{torch_mm}-{py_tag}-{py_tag}-{platform}.whl"
+    wheel_template: "https://github.com/PozzettiAndrea/cuda-wheels/releases/download/cc_torch-latest/cc_torch-{version}%2Bcu{cuda_short}torch{torch_mm}-{py_tag}-{py_tag}-{platform}.whl"
+    default_version: "0.2"
     description: GPU-accelerated connected components
 
   torch_generic_nms:
-    method: index
-    index_url: "https://pozzettiandrea.github.io/cuda-wheels"
-    wheel_template: "https://pozzettiandrea.github.io/cuda-wheels/torch-generic-nms/torch_generic_nms-{version}%2Bcu{cuda_short}torch{torch_mm}-{py_tag}-{py_tag}-{platform}.whl"
+    wheel_template: "https://github.com/PozzettiAndrea/cuda-wheels/releases/download/torch_generic_nms-latest/torch_generic_nms-{version}%2Bcu{cuda_short}torch{torch_mm}-{py_tag}-{py_tag}-{platform}.whl"
+    default_version: "0.1"
     description: GPU-accelerated Non-Maximum Suppression
 
   lietorch:
-    method: index
-    index_url: "https://pozzettiandrea.github.io/cuda-wheels"
-    wheel_template: "https://pozzettiandrea.github.io/cuda-wheels/lietorch/lietorch-{version}%2Bcu{cuda_short}torch{torch_mm}-{py_tag}-{py_tag}-{platform}.whl"
+    wheel_template: "https://github.com/PozzettiAndrea/cuda-wheels/releases/download/lietorch-latest/lietorch-{version}%2Bcu{cuda_short}torch{torch_mm}-{py_tag}-{py_tag}-{platform}.whl"
+    default_version: "0.3"
     description: Lie group operations for PyTorch (DPVO dependency)
 
-  dpvo-cuda:
-    method: index
-    index_url: "https://pozzettiandrea.github.io/cuda-wheels"
-    wheel_template: "https://pozzettiandrea.github.io/cuda-wheels/dpvo_cuda/dpvo_cuda-{version}%2Bcu{cuda_short}torch{torch_mm}-{py_tag}-{py_tag}-{platform}.whl"
-    default_version: "1.0.0"
-    description: DPVO CUDA extensions (cuda_corr, cuda_ba, lietorch_backends)
-
-  # ===========================================================================
-  # 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: index
-    index_url: "https://pozzettiandrea.github.io/cuda-wheels"
+    wheel_template: "https://github.com/PozzettiAndrea/cuda-wheels/releases/download/sageattention-latest/sageattention-{version}%2Bcu{cuda_short}torch{torch_mm}-{py_tag}-{py_tag}-{platform}.whl"
+    default_version: "2.2.0"
     description: SageAttention - 2-5x faster than FlashAttention with quantized kernels
 
-  # ===========================================================================
-  # triton - Required for sageattention on Linux
-  # ===========================================================================
-  triton:
-    method: pypi
-    description: Triton compiler for custom CUDA kernels
+  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
 
   # ===========================================================================
   # detectron2 - Facebook's detection library
   # https://github.com/facebookresearch/detectron2
   # Prebuilt wheels from miropsota's torch_packages_builder
-  # Version format: 0.6+<commit>pt<torch_version>cu<cuda_short>
   # ===========================================================================
   detectron2:
-    method: index
-    index_url: "https://miropsota.github.io/torch_packages_builder"
-    version_template: "0.6+2a420edpt{torch_version}cu{cuda_short}"
+    wheel_template: "https://miropsota.github.io/torch_packages_builder/detectron2/detectron2-0.6%2B2a420edpt{torch_version}cu{cuda_short}-{py_tag}-{py_tag}-{platform}.whl"
     description: Detectron2 - Facebook's detection and segmentation library
 
   # ===========================================================================
-  # flash-attn - Multi-source prebuilt wheels
+  # flash-attn - Dao-AILab (Linux x86_64)
+  # For other platforms, users should add custom wheel_source in their config
   # ===========================================================================
   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]
+    wheel_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"
+    description: Flash Attention for fast transformer inference (Linux x86_64)
+
+  # ===========================================================================
+  # spconv - Sparse convolution library
+  # PyPI with CUDA-versioned package names (spconv-cu124, spconv-cu126, etc.)
+  # Note: This uses a special package_name field, not wheel_template
+  # ===========================================================================
+  spconv:
+    package_name: "spconv-cu{cuda_short2}"
+    description: Sparse convolution library (installs spconv-cu124, spconv-cu126, etc.)