fix

2025-09-11 13:12:08 -04:00
parent f0fee9c56c
commit 47ad92c6b9
2 changed files with 35 additions and 1 deletions
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -106,8 +106,9 @@ dependencies = [
 ]

 [project.optional-dependencies]
-# Specific hardware acceleration
+# Flash attention context parallelism
 ring-flash-attn = [
+    "flash-attn==2.8.3 ; sys_platform == 'linux'",
    "ring-flash-attn>=0.1.7",
    "yunchang==0.6.0",
 ]