coolroman commited on
Commit
70f2888
·
verified ·
1 Parent(s): cab556e

scorevision: push artifact

Browse files
Files changed (1) hide show
  1. chute_config.yml +8 -5
chute_config.yml CHANGED
@@ -2,15 +2,18 @@ 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' 'torch>=2.8'
 
6
 
7
  NodeSelector:
8
  gpu_count: 1
9
- min_vram_gb_per_gpu: 24
 
 
 
10
 
11
  Chute:
12
- timeout_seconds: 300
13
  concurrency: 4
14
- max_instances: 5
15
  scaling_threshold: 0.5
16
- shutdown_after_seconds: 14400
 
2
  from_base: parachutes/python:3.12
3
  run_command:
4
  - pip install --upgrade setuptools wheel
5
+ - pip install huggingface_hub==0.19.4 ultralytics==8.2.40 'torch<2.6' opencv-python-headless onnxruntime-gpu
6
+ set_workdir: /app
7
 
8
  NodeSelector:
9
  gpu_count: 1
10
+ min_vram_gb_per_gpu: 16
11
+ include:
12
+ - "3090"
13
+ - "4090"
14
 
15
  Chute:
16
+ shutdown_after_seconds: 300000
17
  concurrency: 4
18
+ max_instances: 1
19
  scaling_threshold: 0.5