chore: update requirements

.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 .
           

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