限制flash atten的版本

shell-nlp · shell-nlp · commit f7d886213b6c · 2025-09-22T23:48:52.000+08:00
diff --git a/.dockerignore b/.dockerignore
@@ -6,4 +6,5 @@ __pycache__/
 *.egg-info
 logs/
 outputs/
-data/
+data/
+.env
diff --git a/pyproject.toml b/pyproject.toml
@@ -31,7 +31,7 @@ dependencies = [
     "diffusers>=0.35.1",
     #"sqlmodel>=0.0.24",
     "autoawq>=0.2.9",
-    "flash-attn",
+    "flash-attn==2.8.2",
 ]
 
 [tool.uv]
diff --git a/requirements.txt b/requirements.txt
@@ -242,7 +242,7 @@ filelock==3.19.1
     #   vllm
 fire==0.7.1
     # via lmdeploy
-flash-attn==2.8.3
+flash-attn==2.8.2
     # via gpt-server (pyproject.toml)
 flashinfer-python==0.3.1
     # via
diff --git a/uv.lock b/uv.lock
