diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml index cd4b4e3..c77b7c6 100644 --- a/.github/workflows/build.yml +++ b/.github/workflows/build.yml @@ -25,7 +25,7 @@ jobs: body: | | Flash-Attention | Python | PyTorch | CUDA | |-----------------|--------|---------|------| - | 1.0.9, 2.4.3, 2.5.6, 2.5.9, 2.6.3, 2.7.0.post2 | 3.10, 3.11, 3.12 | 2.0.1, 2.1.2, 2.2.2, 2.3.1, 2.4.1, 2.5.1 | 11.8.0, 12.1.1, 12.3.2, 12.4.1 | + | 2.4.3, 2.5.6, 2.6.3, 2.7.0.post2 | 3.10, 3.11, 3.12 | 2.0.1, 2.1.2, 2.2.2, 2.3.1, 2.4.1, 2.5.1 | 11.8.0, 12.1.1, 12.4.1 | build_wheels: name: Build wheels and Upload @@ -34,10 +34,10 @@ jobs: strategy: fail-fast: false matrix: - flash-attn-version: ["1.0.9", "2.4.3", "2.5.6", "2.5.9", "2.6.3", "2.7.0.post2"] + flash-attn-version: ["2.4.3", "2.5.6", "2.6.3", "2.7.0.post2"] 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"] - cuda-version: ["11.8.0", "12.1.1", "12.3.2", "12.4.1"] + 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"