commit change

mosaicml · Nov 15, 2024 · ac8a75d · ac8a75d
1 parent 82af8c7
commit ac8a75d
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/.github/workflows/docker.yaml b/.github/workflows/docker.yaml
@@ -17,11 +17,11 @@ jobs:
     strategy:
       matrix:
         include:
-        - name: "2.5.0_cu124"
-          base_image: mosaicml/pytorch:2.5.0_cu124-python3.11-ubuntu20.04
+        - name: "2.5.1_cu124"
+          base_image: mosaicml/pytorch:2.5.1_cu124-python3.11-ubuntu20.04
           dep_groups: "[all]"
-        - name: "2.5.0_cu124_aws"
-          base_image: mosaicml/pytorch:2.5.0_cu124-python3.11-ubuntu20.04-aws
+        - name: "2.5.1_cu124_aws"
+          base_image: mosaicml/pytorch:2.5.1_cu124-python3.11-ubuntu20.04-aws
           dep_groups: "[all]"
     steps: