Use kraken2-compatible chute config

chute_config.yml

@@ -2,24 +2,18 @@ Image:
   from_base: parachutes/python:3.12
   run_command:
     - pip install --upgrade setuptools wheel
-    - pip install 'numpy>=1.23' 'onnxruntime-gpu>=1.16' 'opencv-python>=4.7' 'pillow>=9.5' 'huggingface_hub>=0.19.4' 'pydantic>=2.0' 'pyyaml>=6.0' 'aiohttp>=3.9'
-    - pip install torch==2.8.0 torchvision==0.23.0 torchaudio==2.8.0 --index-url https://download.pytorch.org/whl/cu128
+    - 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'
+    - pip install torch torchvision
 
 NodeSelector:
   gpu_count: 1
-  min_vram_gb_per_gpu: 16
-  max_hourly_price_per_gpu: 0.5
-
-  exclude:
-    - "5090"
-    - b200
-    - h200
-    - a100
-    - mi300x
+  include:
+    - pro_6000
 
 Chute:
+  tee: true
   timeout_seconds: 900
   concurrency: 4
   max_instances: 5
   scaling_threshold: 0.5
-  shutdown_after_seconds: 288000
+  shutdown_after_seconds: 288000