match the cuda version for 2.4.1 build w/o tmux (#2299)

axolotl-ai-cloud · Jan 30, 2025 · 1063d82 · 1063d82
1 parent ac471a6
commit 1063d82
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml
@@ -125,8 +125,8 @@ jobs:
     strategy:
       matrix:
         include:
-          - cuda: 121
-            cuda_version: 12.1.1
+          - cuda: 124
+            cuda_version: 12.4.1
             python_version: "3.11"
             pytorch: 2.4.1
             axolotl_extras: