Update build.yml

mjun0812 · Oct 26, 2024 · 36853c2 · 36853c2
1 parent 404c6e6
commit 36853c2
Showing 1 changed file with 12 additions and 12 deletions.
diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
@@ -51,21 +51,21 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        flash-attn-version: [2.4.3, 2.5.6, 2.5.9, 2.6.3]
-        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.0]
-        cuda-version: [11.8.0, 12.1.1, 12.4.1]
+        flash-attn-version: ['2.4.3', '2.5.6', '2.5.9', '2.6.3']
+        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.0']
+        cuda-version: ['11.8.0', '12.1.1', '12.4.1']
         exclude:
           # torch < 2.2 does not support Python 3.12
-          - python-version: 3.12
-            torch-version: 2.0.1
-          - python-version: 3.12
-            torch-version: 2.1.2
+          - python-version: '3.12'
+            torch-version: '2.0.1'
+          - python-version: '3.12'
+            torch-version: '2.1.2'
           # torch 2.0.1 does not support CUDA 12.x
-          - torch-version: 2.0.1
-            cuda-version: 12.1.1
-          - torch-version: 2.0.1
-            cuda-version: 12.4.1
+          - torch-version: '2.0.1'
+            cuda-version: '12.1.1'
+          - torch-version: '2.0.1'
+            cuda-version: '12.4.1'
 
     steps:
       - uses: actions/checkout@v4