From d1b13bc0003383bc2d0ef00e28d4a963a60bc9ca Mon Sep 17 00:00:00 2001
From: Junya Morioka <mjun@mjunya.com>
Date: Tue, 4 Feb 2025 12:58:31 +0900
Subject: [PATCH] fix ubuntu version

---
 .github/workflows/build.yml | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
index 2eeb8f1..8c96958 100644
--- a/.github/workflows/build.yml
+++ b/.github/workflows/build.yml
@@ -30,13 +30,14 @@ jobs:
   build_wheels:
     name: Build wheels and Upload
     needs: create_releases
-    runs-on: ubuntu-latest
+    runs-on: ubuntu-22.04
     strategy:
       fail-fast: false
       matrix:
         flash-attn-version: ["2.6.3", "2.7.4.post1"]
         python-version: ["3.10", "3.11", "3.12"]
-        torch-version: ["2.0.1", "2.1.2", "2.2.2", "2.3.1", "2.4.1", "2.5.1", "2.6.0"]
+        torch-version:
+          ["2.0.1", "2.1.2", "2.2.2", "2.3.1", "2.4.1", "2.5.1", "2.6.0"]
         cuda-version: ["12.4.1", "12.6.3"]
         exclude:
           # torch < 2.2 does not support Python 3.12