Image:
  from_base: parachutes/python:3.12
  run_command:
    - pip install --upgrade setuptools wheel
    # CPU onnxruntime pinned to the conformity checker version: identical boxes live and in the offline check
    - pip install 'numpy>=1.23,<3' 'onnxruntime==1.26.0' 'opencv-python-headless>=4.8' 'pillow>=9.5' 'huggingface_hub>=0.24' 'pydantic>=2.0' 'pyyaml>=6.0' 'aiohttp>=3.9'
    - pip install torch==2.8.0 torchvision==0.23.0 --index-url https://download.pytorch.org/whl/cu128

# Chutes only accepts TEE chutes (since 2026-05-12): TEE-capable GPU, one instance
NodeSelector:
  gpu_count: 1
  min_vram_gb_per_gpu: 16
  max_hourly_price_per_gpu: 2
  include:
    - pro_6000

Chute:
  timeout_seconds: 900
  concurrency: 4
  max_instances: 1
  scaling_threshold: 0.5
  shutdown_after_seconds: 288000
  tee: true