Image:
  from_base: parachutes/python:3.12
  run_command:
    - pip install --upgrade setuptools wheel
    - pip install "ultralytics==8.3.222" "opencv-python-headless" "numpy" "pydantic"
    - pip install "tensorflow" "torch==2.7.1" "torchvision==0.22.1" "torch-tensorrt==2.7"
  set_workdir: /app


NodeSelector:
  gpu_count: 1
  min_vram_gb_per_gpu: 16
  include:
    - a100
    - a100_40gb
    - "3090"
    - a40
    - a6000
    - h100
    - l40s
  exclude:
    - "5090"
    - b200
    - h200
    - h20
    - mi300x

Chute:
  timeout_seconds: 900
  concurrency: 4
  max_instances: 5
  scaling_threshold: 0.5