Update Dockerfile

Dockerfile

@@ -35,7 +35,6 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
 RUN pip install --upgrade pip setuptools wheel packaging ninja
 
 # Uninstall any prebuilt flash-attn
-
 ENV CONDA_FORGE=1 \
     CONDA_AUTO_UPDATE_CONDA=false
 
@@ -49,13 +48,13 @@ COPY requirements.txt .
 # Pre-install torch so flash-attn doesn't fail
 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
+RUN pip install "flash-attn<2.8" --no-build-isolation --no-cache-dir
 
 # Copy project code
 COPY . .