Update ci

mila-iqia · Oct 4, 2024 · 678e255 · 678e255
1 parent 7f80881
commit 678e255
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/.github/workflows/cloud-ci.yml b/.github/workflows/cloud-ci.yml
@@ -45,7 +45,7 @@ jobs:
       ARM_TENANT_ID: "${{ secrets.ARM_TENANT_ID }}"
       ARM_SUBSCRIPTION_ID: "${{ secrets.ARM_SUBSCRIPTION_ID }}"
       AZURE_CORE_OUTPUT: none
-      _MULTI_GPUS: "diffusion-gpus,dinov2-giant-gpus,lightning-gpus,llava-gpus,resnet152-ddp-gpus,llm-full-mp-gpus,llm-lora-ddp-gpus,llm-lora-mp-gpus"
+      _MULTI_GPUS: "multigpu"
       _MULTI_NODES: "multinode"
 
     steps:
@@ -113,7 +113,7 @@ jobs:
           case "$gpus" in
             "1g")
               RUN_ON="azure__a100"
-              EXCLUDE="$EXCLUDE,$_MULTI_GPUS,$_MULTI_NODES"
+              EXCLUDE="$EXCLUDE,$_MULTI_GPUS"
               ;;
             "2g")
               RUN_ON="azure__a100_x2"