update vllm version 0.9.0

Signed-off-by: pandyamarut <pandyamarut@gmail.com>
2025-06-05 11:59:15 -07:00
parent deeff579f8
commit 70cd1c8113
5 changed files with 67 additions and 6 deletions
@@ -12,7 +12,7 @@ RUN --mount=type=cache,target=/root/.cache/pip \
    python3 -m pip install --upgrade -r /requirements.txt

 # Install vLLM (switching back to pip installs since issues that required building fork are fixed and space optimization is not as important since caching) and FlashInfer 
-RUN python3 -m pip install vllm==0.8.5 && \
+RUN python3 -m pip install vllm==0.9.0.1 && \
    python3 -m pip install flashinfer -i https://flashinfer.ai/whl/cu121/torch2.3

 # Setup for Option 2: Building the Image with the Model included