comfy-env 0.0.16__tar.gz → 0.0.17__tar.gz

{comfy_env-0.0.16 → comfy_env-0.0.17}/PKG-INFO

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

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

{comfy_env-0.0.16 → comfy_env-0.0.17}/src/comfy_env/install.py

@@ -238,7 +238,11 @@ def _get_install_info(
         index_url = _substitute_template(config.get("index_url", ""), env)
 
         if method == "index":
-            return {"method": method, "description": f"from index {index_url}"}
+            version_info = ""
+            if "version_template" in config:
+                resolved_version = _substitute_template(config["version_template"], env)
+                version_info = f" (version {resolved_version})"
+            return {"method": method, "description": f"from index {index_url}{version_info}"}
         elif method == "github_index":
             return {"method": method, "description": f"from {index_url}"}
         elif method == "find_links":
@@ -277,8 +281,13 @@ def _install_cuda_package(
         if method == "index":
             # PEP 503 index - use pip --extra-index-url
             index_url = _substitute_template(config["index_url"], env)
-            pkg_spec = f"{package}=={version}" if version else package
-            log(f"  Installing {package} from PyG index...")
+            # Check for version_template (e.g., detectron2 with embedded torch/cuda version)
+            if "version_template" in config:
+                resolved_version = _substitute_template(config["version_template"], env)
+                pkg_spec = f"{package}=={resolved_version}"
+            else:
+                pkg_spec = f"{package}=={version}" if version else package
+            log(f"  Installing {package} from index...")
             _pip_install_with_index(pkg_spec, index_url, log)
 
         elif method == "github_index":

{comfy_env-0.0.16 → comfy_env-0.0.17}/src/comfy_env/wheel_sources.yml

@@ -120,6 +120,18 @@ packages:
     method: pypi
     description: Triton compiler for custom CUDA kernels
 
+  # ===========================================================================
+  # 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}"
+    description: Detectron2 - Facebook's detection and segmentation library
+
   # ===========================================================================
   # flash-attn - Multi-source prebuilt wheels
   # ===========================================================================