commit change

mosaicml · Nov 14, 2024 · e842ca1 · e842ca1
1 parent 71618c8
commit e842ca1
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/.github/workflows/pr-cpu.yaml b/.github/workflows/pr-cpu.yaml
@@ -23,7 +23,7 @@ jobs:
         include:
         - name: "cpu-2.4.0"
           pip_deps: "[all-cpu]"
-          container: mosaicml/pytorch:2.4.0_cpu-python3.11-ubuntu20.04
+          container: mosaicml/pytorch:2.5.1_cpu-python3.11-ubuntu20.04
           markers: "not gpu"
           pytest_command: "coverage run -m pytest"
     steps:

diff --git a/.github/workflows/pr-gpu.yaml b/.github/workflows/pr-gpu.yaml
@@ -23,7 +23,7 @@ jobs:
       matrix:
         include:
         - name: "gpu-2.4.0-1"
-          container: mosaicml/llm-foundry:2.4.0_cu124-latest
+          container: mosaicml/pytorch:2.5.1_cu124-python3.11-ubuntu20.04
           markers: "gpu"
           pip_deps: "[all]"
           pytest_command: "coverage run -m pytest"
@@ -51,8 +51,8 @@ jobs:
       fail-fast: false
       matrix:
         include:
-        - name: "gpu-2.4.0-2"
-          container: mosaicml/llm-foundry:2.4.0_cu124-latest
+        - name: "gpu-2.5.0-2"
+          container: mosaicml/pytorch:2.5.1_cu124-python3.11-ubuntu20.04
           markers: "gpu"
           pip_deps: "[all]"
           pytest_command: "coverage run -m pytest"
@@ -81,7 +81,7 @@ jobs:
       matrix:
         include:
         - name: "gpu-2.4.0-4"
-          container: mosaicml/llm-foundry:2.4.0_cu124-latest
+          container: mosaicml/pytorch:2.5.1_cu124-python3.11-ubuntu20.04
           markers: "gpu"
           pip_deps: "[all]"
           pytest_command: "coverage run -m pytest"