comfy-env 0.0.65__py3-none-any.whl → 0.0.67__py3-none-any.whl

{comfy_env-0.0.65.dist-info → comfy_env-0.0.67.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: comfy-env
-Version: 0.0.65
+Version: 0.0.67
 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,7 +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-w>=1.0.0
 Requires-Dist: tomli>=2.0.0; python_version < '3.11'
 Requires-Dist: uv>=0.4.0
 Provides-Extra: dev
@@ -68,8 +68,7 @@ Create a `comfy-env.toml` in your node directory:
 
 ```toml
 [cuda]
-nvdiffrast = "0.4.0"
-pytorch3d = "0.7.9"
+packages = ["nvdiffrast", "pytorch3d"]
 
 [packages]
 requirements = ["transformers>=4.56", "pillow"]
@@ -152,12 +151,6 @@ comfy-env install
 # Dry run (show what would be installed)
 comfy-env install --dry-run
 
-# Resolve wheel URLs without installing
-comfy-env resolve nvdiffrast==0.4.0
-
-# List all packages in the built-in registry
-comfy-env list-packages
-
 # Verify installation
 comfy-env doctor
 ```
@@ -167,11 +160,9 @@ comfy-env doctor
 ### Simple Format (comfy-env.toml)
 
 ```toml
-# CUDA packages (uses built-in registry)
+# CUDA packages from https://pozzettiandrea.github.io/cuda-wheels/
 [cuda]
-nvdiffrast = "0.4.0"
-pytorch3d = "0.7.9"
-torch-scatter = "2.1.2"
+packages = ["nvdiffrast", "pytorch3d", "torch-scatter"]
 
 # Regular pip packages
 [packages]
@@ -185,7 +176,7 @@ requirements = ["transformers>=4.56", "pillow"]
 linux = ["libgl1", "libopengl0"]  # apt packages
 
 [local.cuda]
-nvdiffrast = "0.4.0"
+packages = ["nvdiffrast"]
 
 [local.packages]
 requirements = ["pillow", "numpy"]
@@ -196,82 +187,32 @@ python = "3.10"
 cuda = "12.8"
 
 [myenv.cuda]
-torch-scatter = "2.1.2"
+packages = ["torch-scatter"]
 
 [myenv.packages]
 requirements = ["transformers>=4.56"]
-
-# Custom wheel templates (override built-in registry)
-[wheel_sources]
-my-custom-pkg = "https://my-server.com/my-pkg-{version}+cu{cuda_short}-{py_tag}-{platform}.whl"
-```
-
-## Writing Wheel Templates
-
-### Template Variables
-
-| Variable | Example | Description |
-|----------|---------|-------------|
-| `{version}` | `0.4.0` | Package version |
-| `{cuda_version}` | `12.8` | Full CUDA version |
-| `{cuda_short}` | `128` | CUDA without dot |
-| `{cuda_major}` | `12` | CUDA major only |
-| `{torch_version}` | `2.8.0` | Full PyTorch version |
-| `{torch_mm}` | `28` | PyTorch major.minor no dot |
-| `{torch_dotted_mm}` | `2.8` | PyTorch major.minor with dot |
-| `{py_version}` | `3.10` | Python version |
-| `{py_short}` | `310` | Python without dot |
-| `{py_tag}` | `cp310` | Python wheel tag |
-| `{platform}` | `linux_x86_64` | Platform tag |
-
-### Common Wheel URL Patterns
-
-**Pattern 1: Simple CUDA + Python**
-```
-https://example.com/{package}-{version}+cu{cuda_short}-{py_tag}-{py_tag}-{platform}.whl
-```
-
-**Pattern 2: CUDA + PyTorch**
-```
-https://example.com/{package}-{version}+cu{cuda_short}torch{torch_mm}-{py_tag}-{py_tag}-{platform}.whl
-```
-
-**Pattern 3: GitHub Releases**
-```
-https://github.com/org/repo/releases/download/v{version}/{package}-{version}+cu{cuda_short}-{py_tag}-{platform}.whl
-```
-
-### How to Find the Right Template
-
-1. Download a wheel manually from the source
-2. Look at the filename pattern: `nvdiffrast-0.4.0+cu128torch28-cp310-cp310-linux_x86_64.whl`
-3. Replace values with variables: `nvdiffrast-{version}+cu{cuda_short}torch{torch_mm}-{py_tag}-{py_tag}-{platform}.whl`
-4. Prepend the base URL
-
-### Testing Your Template
-
-```bash
-comfy-env resolve my-package==1.0.0
 ```
 
-This shows the resolved URL without installing.
+## CUDA Wheels Index
 
-### Adding Custom Wheel Sources
+CUDA packages are installed from the [cuda-wheels](https://pozzettiandrea.github.io/cuda-wheels/) index, which provides pre-built wheels for:
 
-If a package isn't in the built-in registry, add it to your `comfy-env.toml`:
+- **PyTorch Geometric**: torch-scatter, torch-cluster, torch-sparse, torch-spline-conv
+- **NVIDIA**: nvdiffrast, pytorch3d, gsplat
+- **Attention**: flash-attn, sageattention
+- **Mesh Processing**: cumesh, cubvh
+- **Others**: spconv, detectron2, lietorch, and more
 
-```toml
-[cuda]
-my-custom-pkg = "1.0.0"
+Wheels are automatically selected based on your GPU, CUDA version, PyTorch version, and Python version.
 
-[wheel_sources]
-my-custom-pkg = "https://my-server.com/my-custom-pkg-{version}+cu{cuda_short}-{py_tag}-{platform}.whl"
-```
+### Supported Configurations
 
-Resolution order:
-1. User's `[wheel_sources]` in comfy-env.toml (highest priority)
-2. Built-in `wheel_sources.yml` registry
-3. Error if not found
+| GPU Architecture | CUDA | PyTorch |
+|-----------------|------|---------|
+| Blackwell (sm_100+) | 12.8 | 2.8+ |
+| Ada/Hopper/Ampere (sm_80+) | 12.8 | 2.8 |
+| Turing (sm_75) | 12.8 | 2.8 |
+| Pascal (sm_60) | 12.4 | 2.4 |
 
 ## API Reference
 
@@ -299,7 +240,7 @@ env = RuntimeEnv.detect()
 print(env)
 # Python 3.10, CUDA 12.8, PyTorch 2.8.0, GPU: NVIDIA GeForce RTX 4090
 
-# Get template variables
+# Get environment variables
 vars_dict = env.as_dict()
 # {'cuda_version': '12.8', 'cuda_short': '128', 'torch_mm': '28', ...}
 ```
@@ -344,16 +285,6 @@ print(get_gpu_summary())
 # GPU 0: NVIDIA GeForce RTX 5090 (sm_120) [Blackwell - CUDA 12.8]
 ```
 
-## Built-in Package Registry
-
-Run `comfy-env list-packages` to see all packages in the built-in registry.
-
-The registry includes:
-- PyTorch Geometric packages (torch-scatter, torch-cluster, torch-sparse)
-- NVIDIA packages (nvdiffrast, pytorch3d, gsplat)
-- Flash Attention (flash-attn)
-- And more
-
 ## License
 
 MIT - see LICENSE file.

comfy_env-0.0.67.dist-info/RECORD

@@ -0,0 +1,32 @@
+comfy_env/__init__.py,sha256=s0RkyKsBlDiSI4ZSwivtWLvYhc8DS0CUEWgFLVJbtLA,2176
+comfy_env/cli.py,sha256=ky7jC8eArKbjgYw9Uy52MKajmtLb16wYblrbEOoiy2U,9599
+comfy_env/errors.py,sha256=q-C3vyrPa_kk_Ao8l17mIGfJiG2IR0hCFV0GFcNLmcI,9924
+comfy_env/install.py,sha256=BO8qbrXKOmNnjjabnYouRUn7vecGxE5Z136Vl0eHNGs,5120
+comfy_env/nodes.py,sha256=nBkG2pESeOt5kXSgTDIHAHaVdHK8-rEueR3BxXWn6TE,4354
+comfy_env/prestartup.py,sha256=_b9QIWsHzkQD5VYdiyn0beHMLoWx18aC4RKB8SlJiGE,2048
+comfy_env/config/__init__.py,sha256=4Guylkb-FV8QxhFwschzpzbr2eu8y-KNgNT3_JOm9jc,403
+comfy_env/config/parser.py,sha256=l1RncyvyvO_ySeiKO13knpfqI5Fb1ShCNQEQQ2lW23g,3933
+comfy_env/config/types.py,sha256=H0m_L4nURv-9M-JZvTcyG9VEcbWzBlppA3Xin26B0jM,1883
+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=iYTvCxmf2M9fxK9cM9RyKQyOSJVkpHyR7GXPm45ZFRI,12656
+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
+comfy_env/pixi/platform/base.py,sha256=iS0ptTTVjXRwPU4qWUdvHI7jteuzxGSjWr5BUQ7hGiU,2453
+comfy_env/pixi/platform/darwin.py,sha256=HK3VkLT6DfesAnIXwx2IaUFHTBclF0xTQnC7azWY6Kc,1552
+comfy_env/pixi/platform/linux.py,sha256=xLp8FEbFqZLQrzIZBI9z3C4g23Ab1ASTHLsXDzsdCoA,2062
+comfy_env/pixi/platform/windows.py,sha256=FCOCgpzGzorY9-HueMlJUR8DxM2eH-cj9iZk6K026Is,10891
+comfy_env/templates/comfy-env-instructions.txt,sha256=ve1RAthW7ouumU9h6DM7mIRX1MS8_Tyonq2U4tcrFu8,1031
+comfy_env/templates/comfy-env.toml,sha256=TBa_8V8gdwoRej4Lb_mkqZ_Ytk_quFTnbwwc1YK_gHE,4190
+comfy_env/workers/__init__.py,sha256=TMVG55d2XLP1mJ3x1d16H0SBDJZtk2kMC5P4HLk9TrA,1073
+comfy_env/workers/base.py,sha256=4ZYTaQ4J0kBHCoO_OfZnsowm4rJCoqinZUaOtgkOPbw,2307
+comfy_env/workers/mp.py,sha256=E_SUvvCd11uCW5dIURET8EVHcgRtu36MPOEjCFDq4Ok,23920
+comfy_env/workers/subprocess.py,sha256=UMhKhaJoSjv2-FWnwQq9TeMWtaDLIs3ajAFTq1ngdJw,57844
+comfy_env/workers/tensor_utils.py,sha256=TCuOAjJymrSbkgfyvcKtQ_KbVWTqSwP9VH_bCaFLLq8,6409
+comfy_env-0.0.67.dist-info/METADATA,sha256=J0u--ky-c7j_vt0j8vu2F3tqZrYS8mi_wTqXjMJkx1M,6946
+comfy_env-0.0.67.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+comfy_env-0.0.67.dist-info/entry_points.txt,sha256=J4fXeqgxU_YenuW_Zxn_pEL7J-3R0--b6MS5t0QmAr0,49
+comfy_env-0.0.67.dist-info/licenses/LICENSE,sha256=E68QZMMpW4P2YKstTZ3QU54HRQO8ecew09XZ4_Vn870,1093
+comfy_env-0.0.67.dist-info/RECORD,,