From 8a4bcacdb204c9cce04ff005d51735bf76b539c1 Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Thu, 17 Jul 2025 00:01:23 -0400 Subject: [PATCH] cu126-torch271 for cloud docker image should be tagged with main-latest (#2935) --- .github/workflows/main.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml index 8692496f1..444ebfde8 100644 --- a/.github/workflows/main.yml +++ b/.github/workflows/main.yml @@ -87,7 +87,6 @@ jobs: python_version: "3.11" pytorch: 2.6.0 axolotl_extras: - is_latest: true - cuda: 126 cuda_version: 12.6.3 python_version: "3.11" @@ -98,6 +97,7 @@ jobs: python_version: "3.11" pytorch: 2.7.1 axolotl_extras: + is_latest: true - cuda: 128 cuda_version: 12.8.1 python_version: "3.11"