v2 fire weights (yolo26n 960 e2e, 3 classes validator-aligned, synth+sim+dfire+z5atr 25k merged, synth val mAP50=0.6395)
479cc25 verified | Image: | |
| from_base: parachutes/python:3.12 | |
| run_command: | |
| - pip install --upgrade setuptools wheel | |
| - pip install 'numpy>=1.23' 'onnxruntime-gpu[cuda,cudnn]>=1.16' 'opencv-python>=4.7' 'pillow>=9.5' 'huggingface_hub>=0.19.4' 'pydantic>=2.0' 'pyyaml>=6.0' 'aiohttp>=3.9' | |
| - pip install torch torchvision | |
| NodeSelector: | |
| gpu_count: 1 | |
| min_vram_gb_per_gpu: 16 | |
| # SN44 chute platform mandates TEE + pro_6000 include for new elements | |
| # (verified by crime + beverage deploys 2026-05-04). Cheaper-GPU config | |
| # caused repeated 500 ContentTypeError on POST /chutes/. | |
| max_hourly_price_per_gpu: 2.00 | |
| include: | |
| - "pro_6000" | |
| exclude: | |
| - "5090" | |
| - b200 | |
| - h200 | |
| - h20 | |
| - mi300x | |
| Chute: | |
| concurrency: 4 | |
| max_instances: 5 | |
| scaling_threshold: 0.5 | |
| shutdown_after_seconds: 288000 # 80h idle | |
| tee: true | |