commit change

mosaicml · May 3, 2024 · 4926b66 · 4926b66
1 parent 62bddd8
commit 4926b66
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/.github/workflows/docker.yaml b/.github/workflows/docker.yaml
@@ -19,10 +19,10 @@ jobs:
         include:
         - name: "2.3.0_cu121_flash2"
           base_image: mosaicml/pytorch:2.3.0_cu121-python3.11-ubuntu20.04
-          dep_groups: "gpu-flash2,te"
+          dep_groups: "[gpu-flash2,te]"
         - name: "2.3.0_cu121_flash2_aws"
           base_image: mosaicml/pytorch:2.3.0_cu121-python3.11-ubuntu20.04-aws
-          dep_groups: "gpu-flash2,te"
+          dep_groups: "[gpu-flash2,te]"
     steps:
     - name: Maximize Build Space on Worker
       uses: easimon/maximize-build-space@v4