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"