diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml index c60a478..bcddd92 100644 --- a/.github/workflows/build.yml +++ b/.github/workflows/build.yml @@ -25,7 +25,7 @@ jobs: body: | | Flash-Attention | Python | PyTorch | CUDA | |-----------------|--------|---------|------| - | 2.4.3, 2.5.9, 2.6.3, 2.7.4.post1 | 3.10, 3.11, 3.12 | 2.4.1, 2.5.1, 2.6.0, 2.7.0 | 11.8.0, 12.4.1, 12.6.3, 12.8.1 | + | 2.4.3, 2.5.9, 2.6.3, 2.7.4.post1 | 3.10, 3.11, 3.12 | 2.4.1, 2.5.1, 2.6.0, 2.7.0 | 11.8.0, 12.4.1, 12.6.3 | build_wheels: name: Build wheels and Upload @@ -39,7 +39,7 @@ jobs: python-version: ["3.10", "3.11", "3.12"] torch-version: ["2.4.1", "2.5.1", "2.6.0", "2.7.0"] # https://developer.nvidia.com/cuda-toolkit-archive - cuda-version: ["11.8.0", "12.4.1", "12.6.3", "12.8.1"] + cuda-version: ["11.8.0", "12.4.1", "12.6.3"] exclude: # torch < 2.2 does not support Python 3.12 - python-version: "3.12"