From c8497c13a0c74a93cc2d7dacef53f1d0f47f7e84 Mon Sep 17 00:00:00 2001 From: Junya Morioka Date: Sat, 2 Nov 2024 07:26:17 +0900 Subject: [PATCH] add Python 3.10, flash-attn v1.0.9 --- .github/workflows/build.yml | 26 +++++--------------------- 1 file changed, 5 insertions(+), 21 deletions(-) diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml index bf0d95c..3e47796 100644 --- a/.github/workflows/build.yml +++ b/.github/workflows/build.yml @@ -23,25 +23,9 @@ jobs: tag_name: ${{ steps.extract_branch.outputs.branch }} release_name: ${{ steps.extract_branch.outputs.branch }} body: | - - Flash-Attention - - 2.4.3 - - 2.5.6 - - 2.5.9 - - 2.6.3 - - Python - - 3.11 - - 3.12 - - PyTorch - - 2.0.1 - - 2.1.2 - - 2.2.2 - - 2.3.1 - - 2.4.1 - - 2.5.0 - - CUDA - - 11.8.0 - - 12.1.1 - - 12.4.1 + | Flash-Attention | Python | PyTorch | CUDA | + |-----------------|--------|---------|------| + | 1.0.9, 2.4.3, 2.5.6, 2.5.9, 2.6.3 | 3.10, 3.11, 3.12 | 2.0.1, 2.1.2, 2.2.2, 2.3.1, 2.4.1, 2.5.0 | 11.8.0, 12.1.1, 12.4.1 | build_wheels: name: Build wheels and Upload @@ -50,8 +34,8 @@ jobs: strategy: fail-fast: false matrix: - flash-attn-version: ["2.4.3", "2.5.6", "2.5.9", "2.6.3"] - python-version: ["3.11", "3.12"] + flash-attn-version: ["1.0.9", "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: