fix

Author: speed1313

pyproject.toml

@@ -54,21 +54,17 @@ allow-direct-references = true
 
 [dependency-groups]
 dev = [
-    "flash-attn>=2.7.3",
-    "pytest>=8.3.4",
-    "seaborn>=0.13.2",
-]
-
-test = [
     "pytest>=8.3.4",
     "seaborn>=0.13.2",
 ]
 
 evovlm = [
+    "flash-attn>=2.7.3",
     "transformers==4.42.4",
     "mantis-vl",
 ]
 vilaja = [
+    "flash-attn>=2.7.3",
     "accelerate==0.27.2",
     "deepspeed>=0.16.3",
     "einops>=0.8.0",
@@ -79,6 +75,7 @@ vilaja = [
     "transformers==4.37.2",
 ]
 sarashina = [
+    "flash-attn>=2.7.3",
     "accelerate>=0.27.2",
     "pillow>=10.4.0",
     "protobuf>=5.29.3",
@@ -88,10 +85,12 @@ sarashina = [
     "transformers==4.47.0",
 ]
 gemma = [
+    "flash-attn>=2.7.3",
     "accelerate>=0.27.2",
     "timm==1.0.13",
 ]
 normal = [
+    "flash-attn>=2.7.3",
     "accelerate>=1.2.1",
     "qwen-vl-utils>=0.0.8",
     "sentencepiece>=0.2.0",
@@ -100,6 +99,7 @@ normal = [
     "transformers>=4.46.3",
 ]
 pixtral = [
+    "flash-attn>=2.7.3",
     "mistral-common==1.5.0",
     "vllm==0.6.5",
 ]