update names

mosaicml · Feb 12, 2024 · 2413022 · 2413022
1 parent 1bcf894
commit 2413022
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/.github/workflows/docker.yaml b/.github/workflows/docker.yaml
@@ -17,10 +17,10 @@ jobs:
     strategy:
       matrix:
         include:
-        - name: "2.1.0_cu121_flash2"
+        - name: "2.1.2_cu121_flash2"
           base_image: mosaicml/pytorch:2.1.2_cu121-python3.10-ubuntu20.04
           dep_groups: "[gpu-flash2]"
-        - name: "2.1.0_cu121_flash2_aws"
+        - name: "2.1.2_cu121_flash2_aws"
           base_image: mosaicml/pytorch:2.1.2_cu121-python3.10-ubuntu20.04-aws
           dep_groups: "[gpu-flash2]"
     steps: