Update Dockerfile

Dockerfile

@@ -49,13 +49,6 @@ COPY requirements.txt .
 RUN pip install --upgrade pip && \
     pip install torch==2.6 torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu121
     
-# Install the rest of the dependencies
-RUN pip install --no-cache-dir -r requirements.txt
-RUN pip uninstall -y flash-attn || true
-
-# Build from source so it links correctly
-RUN pip install "flash-attn<2.8" --no-build-isolation --no-cache-dir
-
 # Copy project code
 COPY . .