update pyproject.toml

Author: pan-x-c

README.md

@@ -148,8 +148,11 @@ pip install -e .\[dev\]
 
 # Install flash-attn after all dependencies are installed
 # Note: flash-attn will take a long time to compile, please be patient.
-pip install flash-attn -v
-# Try the following command if you encounter errors during installation
+# for bash
+pip install -e .[flash_attn]
+# for zsh
+pip install -e .\[flash_attn\]
+# Try the following command if you encounter errors during flash-attn installation
 # pip install flash-attn -v --no-build-isolation
 ```
 

pyproject.toml

@@ -25,7 +25,6 @@ dependencies = [
     "ray[default]>=2.45.0",
     "vllm==0.9.1",
     "tensordict==0.6.2",
-    "flash-attn==2.8.0.post2",
     "wandb",
     "omegaconf",
     "sqlalchemy",
@@ -70,6 +69,10 @@ doc = [
     "myst-parser",
 ]
 
+flash_attn = [
+    "flash-attn==2.8.0.post2"
+]
+
 [tool.setuptools.packages.find]
 where = ["."]
 include = ["trinity*"]