From 5c629ee4447b64b77f465bc14ded44d37efdfa9b Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Mon, 21 Oct 2024 19:51:06 -0400 Subject: [PATCH] use torch 2.4.1 images as latest now that torch 2.5.0 is out (#1987) --- .github/workflows/main.yml | 4 ++-- .github/workflows/nightlies.yml | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml index 47a4c7f11..3b82f6a51 100644 --- a/.github/workflows/main.yml +++ b/.github/workflows/main.yml @@ -23,12 +23,12 @@ 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" pytorch: 2.4.1 axolotl_extras: + is_latest: true - cuda: 124 cuda_version: 12.4.1 python_version: "3.11" @@ -85,12 +85,12 @@ 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" pytorch: 2.4.1 axolotl_extras: + is_latest: true - cuda: 124 cuda_version: 12.4.1 python_version: "3.11" diff --git a/.github/workflows/nightlies.yml b/.github/workflows/nightlies.yml index 55123a902..b2110e737 100644 --- a/.github/workflows/nightlies.yml +++ b/.github/workflows/nightlies.yml @@ -22,12 +22,12 @@ 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" pytorch: 2.4.1 axolotl_extras: + is_latest: true - cuda: 124 cuda_version: 12.4.1 python_version: "3.11" @@ -84,12 +84,12 @@ 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" pytorch: 2.4.1 axolotl_extras: + is_latest: true - cuda: 124 cuda_version: 12.4.1 python_version: "3.11"