fix

huggingface · Sep 23, 2024 · ef9f3c4 · ef9f3c4
1 parent 41fbbe0
commit ef9f3c4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/test_cli_cuda_tensorrt_llm.yaml b/.github/workflows/test_cli_cuda_tensorrt_llm.yaml
@@ -59,7 +59,7 @@ jobs:
       contains( github.event.pull_request.labels.*.name, 'cli_cuda_tensorrt_llm_multi_gpu')
       }}
 
-    runs-on: [multi-gpu, nvidia-gpu, a10, ci]
+    runs-on: [multi-gpu, nvidia-gpu, 4-a10, ci]
 
     container:
       image: huggingface/optimum-nvidia:latest