From 2888171ba2fed5a5d58242547aa1a6a69f6bd594 Mon Sep 17 00:00:00 2001
From: Rachneet <rachneetsachdeva@yahoo.com>
Date: Fri, 3 Jan 2025 13:06:29 +0100
Subject: [PATCH] chore: update libraries

---
 .github/workflows/tests.yml | 1 +
 requirements.txt            | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

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