chore: update libraries

UKPLab · Jan 3, 2025 · 2888171 · 2888171
1 parent 6d167a7
commit 2888171
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git 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
@@ -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