ultravision-01 / config.yml
mohantesting's picture
Add files using upload-large-folder tool
f60a6c1 verified
raw
history blame contribute delete
601 Bytes
Image:
from_base: parachutes/python:3.12
run_command:
- pip install --upgrade setuptools wheel
- pip install huggingface_hub==0.19.4 ultralytics==8.3.224 'torch<2.6' opencv-python-headless
- pip install scikit-learn
- pip install boto3
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
- mi300x
Chute:
timeout_seconds: 300
concurrency: 4
max_instances: 5
scaling_threshold: 0.5