scorevision: push artifact

chute_config.yml

@@ -2,19 +2,28 @@ Image:
   from_base: parachutes/python:3.12
   run_command:
     - pip install --upgrade setuptools wheel
-    - pip install huggingface_hub==0.19.4 'onnxruntime-gpu[cuda,cudnn]>=1.18' opencv-python-headless numpy pydantic pyyaml aiohttp
-    - pip install 'torch<2.6'
+    - pip install huggingface_hub==0.19.4 opencv-python-headless numpy pydantic pyyaml aiohttp
+    - pip install --index-url https://download.pytorch.org/whl/cu128 torch==2.8.0
+    - pip install 'onnxruntime-gpu>=1.20,<1.25'
   set_workdir: /app
 
 NodeSelector:
   gpu_count: 1
   min_vram_gb_per_gpu: 16
-  exclude:
-    - "5090"
-    - b200
-    - h200
-    - h20
-    - mi300x
+  include:
+    - "4090"
+    - a6000
+    - a6000_ada
+    - l40s
+    - l40
+    - a100
+    - a100_40gb
+    - a100_40gb_sxm
+    - a100_sxm
+    - h100
+    - h100_sxm
+    - h100_nvl
+    - h800
 
 Chute:
   shutdown_after_seconds: 300