From ef9f3c46941ed6fd77d9299a84884efafab1072b Mon Sep 17 00:00:00 2001 From: IlyasMoutawwakil Date: Mon, 23 Sep 2024 13:53:21 +0200 Subject: [PATCH] fix --- .github/workflows/test_cli_cuda_tensorrt_llm.yaml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/test_cli_cuda_tensorrt_llm.yaml b/.github/workflows/test_cli_cuda_tensorrt_llm.yaml index 2fbcc0b7..8f432364 100644 --- 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