From e842ca104e57446c3fe4cdc88df1b5687335b41c Mon Sep 17 00:00:00 2001 From: Chuck Tang Date: Thu, 14 Nov 2024 15:40:43 -0800 Subject: [PATCH] commit change --- .github/workflows/pr-cpu.yaml | 2 +- .github/workflows/pr-gpu.yaml | 8 ++++---- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/.github/workflows/pr-cpu.yaml b/.github/workflows/pr-cpu.yaml index 056b070143..9bc76bc206 100644 --- 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 index 5b91d54442..c709ecf78a 100644 --- 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"