diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
index 564073d..140eb6a 100644
--- a/.github/workflows/tests.yml
+++ b/.github/workflows/tests.yml
@@ -31,6 +31,7 @@ jobs:
       - name: Install dependencies
         run: |
           pip install torch==2.4.0
+          pip install flash-attn==2.6.3 --no-build-isolation
           pip install .
           pip install -r requirements-dev.txt
       
diff --git a/requirements.txt b/requirements.txt
index f910e24..41df641 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -6,7 +6,7 @@ fastapi==0.111.0
 fastapi-cli==0.0.4
 fastchat==0.1.0
 torch==2.4.0
-flash-attn==2.6.3 --install-option="--no-build-isolation"
+flash-attn==2.6.3
 flashinfer==0.1.2+cu121torch2.3
 huggingface-hub==0.23.4
 langchain==0.3.3