Update Dockerfile

Dockerfile

@@ -61,7 +61,7 @@ RUN pip install uv setuptools
 
 # Install vLLM
 # 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/cu113
+RUN uv pip install --system --index-strategy unsafe-best-match vllm==0.10.0 --extra-index-url https://download.pytorch.org/whl/cu128
 
 # # Then, install xformers with the --no-build-isolation flag
 # RUN uv pip install --system \