Update Dockerfile

Dockerfile

@@ -21,10 +21,10 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
 WORKDIR /app
 COPY requirements.txt /app/requirements.txt
 
-# Torch (CUDA 12.1) first, then other deps (excluding deepspeed)
+# Torch first, then everything except deepspeed / flash-attn
 RUN python3 -m pip install --upgrade pip \
  && python3 -m pip install --extra-index-url https://download.pytorch.org/whl/cu121 torch torchvision torchaudio \
- && awk '!/^ *#/ && !/deepspeed/ {print $0}' requirements.txt > requirements.infer.txt \
+ && awk '!/^ *#/ && !/deepspeed/ && !/flash-attn/ && !/flash_attn/ {print $0}' requirements.txt > requirements.infer.txt \
  && python3 -m pip install -r requirements.infer.txt \
  && python3 -m pip install huggingface_hub imageio-ffmpeg