From d8b4027200de0fe60f4ae0a71272c1a8cb2888f7 Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Fri, 10 Jan 2025 08:35:25 -0500 Subject: [PATCH] use 2.5.1 docker images as latest tag as it seems stable (#2198) --- .github/workflows/main.yml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml index b4344dfe2..89b2746e4 100644 --- a/.github/workflows/main.yml +++ b/.github/workflows/main.yml @@ -25,7 +25,6 @@ jobs: python_version: "3.11" pytorch: 2.3.1 axolotl_extras: mamba-ssm - is_latest: true - cuda: 124 cuda_version: 12.4.1 python_version: "3.11" @@ -36,6 +35,7 @@ jobs: python_version: "3.11" pytorch: 2.5.1 axolotl_extras: + is_latest: true runs-on: axolotl-gpu-runner steps: - name: Checkout @@ -92,7 +92,6 @@ jobs: python_version: "3.11" pytorch: 2.3.1 axolotl_extras: - is_latest: true - cuda: 124 cuda_version: 12.4.1 python_version: "3.11" @@ -103,6 +102,7 @@ jobs: python_version: "3.11" pytorch: 2.5.1 axolotl_extras: + is_latest: true runs-on: axolotl-gpu-runner steps: - name: Checkout