Update Dockerfile

Dockerfile

@@ -57,12 +57,12 @@ RUN python3 --version && python3 -m pip --version
 # RUN pip install --upgrade pip
 
 # Install uv, setuptools, and torch for xformers
-RUN pipx install uv && pipx ensurepath
-
-
 # Install vLLM
+RUN pipx install uv && pipx ensurepath && \
+  export PATH="/root/.local/bin:$PATH" && \
+  uv pip install --system --index-strategy unsafe-best-match vllm==0.10.0 --extra-index-url https://download.pytorch.org/whl/cu128
+
 # RUN uv pip install --system vllm==0.10.0 torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113
-RUN uv pip install --system --index-strategy unsafe-best-match vllm==0.10.0 --extra-index-url https://download.pytorch.org/whl/cu128
 
 
 # Downgrade triton because following error occured when using triton==3.3.1