| Image: | |
| from_base: parachutes/python:3.12 | |
| set_workdir: /app | |
| run_command: | |
| - pip install --upgrade setuptools wheel | |
| - pip install --no-cache-dir torch==2.8.0 torchvision==0.23.0 | |
| - pip install --no-cache-dir onnxruntime opencv-python-headless huggingface_hub pyyaml aiohttp pydantic numpy pillow | |
| - git clone https://github.com/BluxWarrior/yolo_onnx_v2 | |
| - python /app/yolo_onnx_v2/install_ort_patch.py | |
| - python -c "import onnxruntime as ort, inspect; print('ORT', ort.__version__); print('Session', ort.InferenceSession); print('File', inspect.getfile(ort.InferenceSession))" | |
| NodeSelector: | |
| gpu_count: 1 | |
| min_vram_gb_per_gpu: 16 | |
| max_hourly_price_per_gpu: 2 | |
| include: | |
| - pro_6000 | |
| Chute: | |
| concurrency: 4 | |
| max_instances: 5 | |
| scaling_threshold: 0.5 | |
| shutdown_after_seconds: 288000 | |
| tee: true |