From 1063d82b512d61cddab87d3faf5c4a5ef6130a3b Mon Sep 17 00:00:00 2001
From: Wing Lian <wing.lian@gmail.com>
Date: Thu, 30 Jan 2025 11:46:09 -0500
Subject: [PATCH] match the cuda version for 2.4.1 build w/o tmux (#2299)

---
 .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 4f8074ad1..c9e13957c 100644
--- 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: