nexu02 commited on
Commit
2b100f7
·
verified ·
1 Parent(s): 1ca3e6c

scorevision: push artifact

Browse files
Files changed (1) hide show
  1. chute_config.yml +2 -2
chute_config.yml CHANGED
@@ -2,8 +2,8 @@ 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
 
2
  from_base: parachutes/python:3.12
3
  run_command:
4
  - pip install --upgrade setuptools wheel
5
+ - pip install 'ultralytics>=8.4' 'torch>=2.4' 'torchvision' 'opencv-python-headless' 'pillow>=9.5' 'huggingface_hub>=0.19.4' 'pydantic>=2.0' 'numpy>=1.23'
6
+ set_workdir: /app
7
 
8
  NodeSelector:
9
  gpu_count: 1