comfy-env 0.0.74__py3-none-any.whl → 0.0.75__py3-none-any.whl

comfy_env/pixi/core.py

@@ -31,6 +31,12 @@ PIXI_URLS = {
 # CUDA wheels index
 CUDA_WHEELS_INDEX = "https://pozzettiandrea.github.io/cuda-wheels/"
 
+# Flash attention pre-built wheels (mjun0812)
+FLASH_ATTN_INDEX = "https://github.com/mjun0812/flash-attention-prebuild-wheels/releases/expanded_assets/v0.7.16"
+
+# PyTorch Geometric wheels index
+PYG_WHEELS_INDEX = "https://data.pyg.org/whl/"
+
 # CUDA version -> PyTorch version mapping
 CUDA_TORCH_MAP = {
     "12.8": "2.8",
@@ -39,6 +45,20 @@ CUDA_TORCH_MAP = {
 }
 
 
+def get_all_find_links(package: str, torch_version: str, cuda_version: str) -> list:
+    """Get all find-links URLs for a CUDA package."""
+    cuda_short = cuda_version.replace(".", "")[:3]
+
+    return [
+        # cuda-wheels index (package-specific page)
+        f"{CUDA_WHEELS_INDEX}{package}/",
+        # flash-attn GitHub releases
+        FLASH_ATTN_INDEX,
+        # PyG index (torch version specific)
+        f"{PYG_WHEELS_INDEX}torch-{torch_version}.0+cu{cuda_short}.html",
+    ]
+
+
 def get_current_platform() -> str:
     """Get the current platform string for pixi."""
     if sys.platform == "linux":
@@ -310,7 +330,7 @@ def pixi_install(
         log(f"pixi install failed:\n{result.stderr}")
         raise RuntimeError(f"pixi install failed: {result.stderr}")
 
-    # Install CUDA packages with --no-index --find-links (bypasses PyPI completely)
+    # Install CUDA packages with --find-links (searches all known sources)
     if cfg.cuda_packages and cuda_version:
         log(f"Installing CUDA packages: {cfg.cuda_packages}")
         python_path = get_pixi_python(node_dir)
@@ -318,18 +338,18 @@ def pixi_install(
             raise RuntimeError("Could not find Python in pixi environment")
 
         for package in cfg.cuda_packages:
-            # Each package has its own find-links page at CUDA_WHEELS_INDEX/<package>/
-            find_links_url = f"{CUDA_WHEELS_INDEX}{package}/"
-            log(f"  Installing {package} from {find_links_url}")
-
             pip_cmd = [
                 str(python_path), "-m", "pip", "install",
-                "--no-index",
                 "--no-deps",
                 "--no-cache-dir",
-                "--find-links", find_links_url,
-                package,
             ]
+
+            # Add all find-links sources
+            for url in get_all_find_links(package, torch_version, cuda_version):
+                pip_cmd.extend(["--find-links", url])
+
+            log(f"  Installing {package}")
+            pip_cmd.append(package)
             result = subprocess.run(pip_cmd, capture_output=True, text=True)
             if result.returncode != 0:
                 log(f"CUDA package install failed for {package}:\n{result.stderr}")

{comfy_env-0.0.74.dist-info → comfy_env-0.0.75.dist-info}/METADATA

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

@@ -10,7 +10,7 @@ comfy_env/config/types.py,sha256=Sb8HO34xsSZu5YAc2K4M7Gb3QNevJlngf12hHiwuU0w,214
 comfy_env/isolation/__init__.py,sha256=vw9a4mpJ2CFjy-PLe_A3zQ6umBQklgqWNxwn9beNw3g,175
 comfy_env/isolation/wrap.py,sha256=9bXxK9h4FMrFL9k4EAdILWxkjXYytVCJV68MuwwufK8,11485
 comfy_env/pixi/__init__.py,sha256=BUrq7AQf3WDm0cHWh72B2xZbURNnDu2dCuELWiQCUiM,997
-comfy_env/pixi/core.py,sha256=Z6PEALzruOJnUvW0fH2n8BeCsCeaeDDrerY6TnhMPxs,12072
+comfy_env/pixi/core.py,sha256=BAaO-6a4XmnnNXZbfMla7egBNKSqgX9cSXHVQsQP6xI,12752
 comfy_env/pixi/cuda_detection.py,sha256=sqB3LjvGNdV4eFqiARQGfyecBM3ZiUmeh6nG0YCRYQw,9751
 comfy_env/pixi/resolver.py,sha256=U_A8rBDxCj4gUlJt2YJQniP4cCKqxJEiVFgXOoH7vM8,6339
 comfy_env/pixi/platform/__init__.py,sha256=Nb5MPZIEeanSMEWwqU4p4bnEKTJn1tWcwobnhq9x9IY,614
@@ -25,8 +25,8 @@ comfy_env/workers/base.py,sha256=4ZYTaQ4J0kBHCoO_OfZnsowm4rJCoqinZUaOtgkOPbw,230
 comfy_env/workers/mp.py,sha256=d6PFVrgqp3MK7Gkt08a8LQD_VSwHtoIcGx2Lovou3vM,25972
 comfy_env/workers/subprocess.py,sha256=UMhKhaJoSjv2-FWnwQq9TeMWtaDLIs3ajAFTq1ngdJw,57844
 comfy_env/workers/tensor_utils.py,sha256=TCuOAjJymrSbkgfyvcKtQ_KbVWTqSwP9VH_bCaFLLq8,6409
-comfy_env-0.0.74.dist-info/METADATA,sha256=rmFALPp_CocGIiAe5vF03FMHrJvFtrSt27UhPGVmmUc,6946
-comfy_env-0.0.74.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-comfy_env-0.0.74.dist-info/entry_points.txt,sha256=J4fXeqgxU_YenuW_Zxn_pEL7J-3R0--b6MS5t0QmAr0,49
-comfy_env-0.0.74.dist-info/licenses/LICENSE,sha256=E68QZMMpW4P2YKstTZ3QU54HRQO8ecew09XZ4_Vn870,1093
-comfy_env-0.0.74.dist-info/RECORD,,
+comfy_env-0.0.75.dist-info/METADATA,sha256=UoNQP1R_ddXCEFsPIpALxZp3wGBKKnj9xmJWU8guiis,6946
+comfy_env-0.0.75.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+comfy_env-0.0.75.dist-info/entry_points.txt,sha256=J4fXeqgxU_YenuW_Zxn_pEL7J-3R0--b6MS5t0QmAr0,49
+comfy_env-0.0.75.dist-info/licenses/LICENSE,sha256=E68QZMMpW4P2YKstTZ3QU54HRQO8ecew09XZ4_Vn870,1093
+comfy_env-0.0.75.dist-info/RECORD,,