From 641db759ab7168e472909bc9ff1eda4a329de34f Mon Sep 17 00:00:00 2001 From: Tri Dao Date: Tue, 12 Nov 2024 20:01:43 -0800 Subject: [PATCH] [CI] Pytorch 2.5.1 does not support python 3.8 --- .github/workflows/publish.yml | 3 +++ flash_attn/__init__.py | 2 +- 2 files changed, 4 insertions(+), 1 deletion(-) diff --git a/.github/workflows/publish.yml b/.github/workflows/publish.yml index 1cfe08b25..d9789739a 100644 --- a/.github/workflows/publish.yml +++ b/.github/workflows/publish.yml @@ -56,6 +56,9 @@ jobs: # Pytorch < 2.2 does not support Python 3.12 - torch-version: '2.1.2' python-version: '3.12' + # Pytorch >= 2.5 does not support Python 3.8 + - torch-version: '2.5.1' + python-version: '3.8' steps: - name: Checkout diff --git a/flash_attn/__init__.py b/flash_attn/__init__.py index 7a784dfee..aa46ec2e0 100644 --- a/flash_attn/__init__.py +++ b/flash_attn/__init__.py @@ -1,4 +1,4 @@ -__version__ = "2.7.0.post1" +__version__ = "2.7.0.post2" from flash_attn.flash_attn_interface import ( flash_attn_func,