From 18ce28586e97a64293455be11479e9216495d186 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?oliver=20k=C3=B6nig?= Date: Fri, 13 Dec 2024 23:55:39 +0100 Subject: [PATCH] fix (#11575) Signed-off-by: Oliver Koenig --- .github/workflows/cicd-main.yml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/cicd-main.yml b/.github/workflows/cicd-main.yml index 74f20ed52392..3008ca6d2435 100644 --- a/.github/workflows/cicd-main.yml +++ b/.github/workflows/cicd-main.yml @@ -4027,7 +4027,7 @@ jobs: uses: ./.github/workflows/_test_template.yml if: contains(fromJSON(needs.cicd-test-container-setup.outputs.test_to_run), 'L2_NeMo_2_SSM_Pretraining') || needs.cicd-test-container-setup.outputs.all == 'true' with: - RUNNER: self-hosted-azure + RUNNER: self-hosted-azure-gpus-1 SCRIPT: | python tests/collections/llm/gpt/model/megatron_ssm_pretraining.py \ @@ -4041,7 +4041,7 @@ jobs: uses: ./.github/workflows/_test_template.yml if: contains(fromJSON(needs.cicd-test-container-setup.outputs.test_to_run), 'L2_NeMo_2_SSM_Finetuning') || needs.cicd-test-container-setup.outputs.all == 'true' with: - RUNNER: self-hosted-azure + RUNNER: self-hosted-azure-gpus-1 SCRIPT: | python tests/collections/llm/gpt/model/megatron_ssm_finetuning.py \