earthboypirl commited on
Commit
4edc6a0
·
verified ·
1 Parent(s): 81b8c63

scorevision: push artifact

Browse files
Files changed (1) hide show
  1. chute_config.yml +7 -9
chute_config.yml CHANGED
@@ -2,24 +2,22 @@ Image:
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: 16
11
- max_hourly_price_per_gpu: 0.5
12
-
13
  exclude:
14
- - "5090"
15
  - b200
16
  - h200
17
- - a100
18
  - mi300x
19
 
20
  Chute:
21
- timeout_seconds: 900
22
  concurrency: 4
23
- max_instances: 1
24
  scaling_threshold: 0.5
25
- shutdown_after_seconds: 300
 
2
  from_base: parachutes/python:3.12
3
  run_command:
4
  - pip install --upgrade setuptools wheel
5
+ # Versions must match the training env (ultralytics 8.4.x / torch 2.x):
6
+ # a model saved by 8.4 will not reliably load under the template's old 8.2.40 pin.
7
+ - pip install huggingface_hub ultralytics==8.4.112 torch opencv-python-headless
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
+ - h20
17
  - mi300x
18
 
19
  Chute:
20
+ shutdown_after_seconds: 300
21
  concurrency: 4
22
+ max_instances: 5
23
  scaling_threshold: 0.5