Revert to flash_attn>=2.7.4.post1 until the wheel situation is sorted out

turboderp · turboderp · commit 327d1f99d691 · 2025-07-16T19:12:46.000+02:00
diff --git a/requirements.txt b/requirements.txt
@@ -1,5 +1,5 @@
 torch>=2.6.0
-flash_attn==2.7.4.post1
+flash_attn>=2.7.4.post1
 tokenizers>=0.21.1
 numpy>=2.1.0
 rich
diff --git a/setup.py b/setup.py
@@ -87,7 +87,7 @@
     author="turboderp",
     install_requires=[
         "torch>=2.6.0",
-        "flash_attn==2.7.4.post1",
+        "flash_attn>=2.7.4.post1",
         "tokenizers>=0.21.1",
         "numpy>=2.1.0",
         "rich",
