iotaminer commited on
Commit
1d9c810
·
verified ·
1 Parent(s): b52a3e2

scorevision: push artifact

Browse files
Files changed (1) hide show
  1. chute_config.yml +7 -12
chute_config.yml CHANGED
@@ -2,22 +2,17 @@ Image:
2
  from_base: parachutes/python:3.12
3
  run_command:
4
  - pip install --upgrade setuptools wheel
5
- - pip install huggingface_hub==0.19.4 opencv-python-headless numpy pydantic pyyaml aiohttp
6
- - pip install --index-url https://download.pytorch.org/whl/cu128 torch==2.8.0
7
- - pip install 'onnxruntime-gpu>=1.20,<1.25'
8
- set_workdir: /app
9
 
10
  NodeSelector:
11
  gpu_count: 1
12
- min_vram_gb_per_gpu: 16
13
- exclude:
14
- - b200
15
- - h200
16
- - mi300x
17
- - b300
18
 
19
  Chute:
20
- shutdown_after_seconds: 604800
21
  concurrency: 4
22
- max_instances: 1
23
  scaling_threshold: 0.5
 
 
2
  from_base: parachutes/python:3.12
3
  run_command:
4
  - pip install --upgrade setuptools wheel
5
+ - 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'
6
+ - pip install torch torchvision
 
 
7
 
8
  NodeSelector:
9
  gpu_count: 1
10
+ min_vram_gb_per_gpu: 8
11
+ max_hourly_price_per_gpu: 1
 
 
 
 
12
 
13
  Chute:
14
+ timeout_seconds: 900
15
  concurrency: 4
16
+ max_instances: 5
17
  scaling_threshold: 0.5
18
+ shutdown_after_seconds: 288000