Update Dockerfile

Dockerfile

@@ -22,6 +22,10 @@ ENV VLLM_USE_TRITON_FLASH_ATTN=0
 # Choose XFORMERS that most stable for T4
 ENV VLLM_ATTENTION_BACKEND=XFORMERS
 
+# Set environment variables for the xformers build
+ENV TORCH_CUDA_ARCH_LIST='7.0 7.5 8.0 8.9 9.0 10.0+PTX'
+ENV MAX_JOBS=16
+
 # Set environment variables to avoid interactive prompts
 ENV DEBIAN_FRONTEND=noninteractive
 
@@ -51,6 +55,13 @@ RUN python3 --version && python3 -m pip --version
 # RUN apt-get update && apt-get install -y python3 python3-pip git
 # RUN pip install --upgrade pip
 
+# Install uv using pip
+RUN pip install uv
+
+# Install xformers using uv
+RUN uv pip install --system \
+    --no-build-isolation "git+https://github.com/facebookresearch/xformers@v0.0.31"
+
 # Install vLLM
 RUN pip install vllm==0.10.0 --extra-index-url https://download.pytorch.org/whl/cu113