From 739d454ed23678824fa2cba0ef3a3eb401b41bb2 Mon Sep 17 00:00:00 2001
From: Rachneet <rachneetsachdeva@yahoo.com>
Date: Fri, 3 Jan 2025 13:25:09 +0100
Subject: [PATCH] chore: update requirements

---
 .github/workflows/tests.yml | 19 -------------------
 requirements.txt            |  1 -
 2 files changed, 20 deletions(-)

diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
index a6cada0..5d01d41 100644
--- a/.github/workflows/tests.yml
+++ b/.github/workflows/tests.yml
@@ -28,27 +28,8 @@ jobs:
         with: 
           python-version: 3.9
 
-      - name: Install system dependencies
-        run: |
-          sudo apt-get update
-          sudo apt-get install -y build-essential
-          sudo apt-get install -y cuda-toolkit-11.2  # Adjust based on your CUDA version
-          sudo apt-get install -y libomp-dev  # Optional for parallel builds
-
       - name: Install Python dependencies
         run: |
-          # Install numpy before torch to avoid missing numpy error
-          pip install numpy
-          
-          # Install PyTorch with the specific version (with CUDA 11.2 support here)
-          pip install torch==2.4.0+cu121  # Modify for the correct CUDA version
-          
-          # Install packaging, wheel, and other required libraries
-          pip install packaging wheel
-          
-          # Install flash-attn with no build isolation
-          pip install flash-attn==2.6.3 --no-build-isolation
-          
           # Install the current package (from setup.py)
           pip install .
           
diff --git a/requirements.txt b/requirements.txt
index 41df641..8f009e7 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -6,7 +6,6 @@ fastapi==0.111.0
 fastapi-cli==0.0.4
 fastchat==0.1.0
 torch==2.4.0
-flash-attn==2.6.3
 flashinfer==0.1.2+cu121torch2.3
 huggingface-hub==0.23.4
 langchain==0.3.3