File tree Expand file tree Collapse file tree 1 file changed +6
-5
lines changed
Expand file tree Collapse file tree 1 file changed +6
-5
lines changed Original file line number Diff line number Diff line change 11ARG NIGHTLY_DATE="20250714"
22ARG BASE_IMAGE="us-central1-docker.pkg.dev/tpu-pytorch-releases/docker/xla:nightly_3.12_tpuvm_$NIGHTLY_DATE"
33# The latest main will be used if arg unspecified
4- ARG VLLM_COMMIT_HASH=""
4+ # ARG VLLM_COMMIT_HASH=""
55
66FROM $BASE_IMAGE
77
@@ -18,11 +18,12 @@ RUN apt-get update && apt-get install -y \
1818# Build vLLM
1919WORKDIR /workspace/vllm
2020ARG VLLM_REPO=https://github.com/vllm-project/vllm.git
21- ARG VLLM_COMMIT_HASH
21+ # ARG VLLM_COMMIT_HASH
2222RUN git clone $VLLM_REPO /workspace/vllm
23- RUN if [ -n "$VLLM_COMMIT_HASH" ]; then \
24- git checkout $VLLM_COMMIT_HASH; \
25- fi
23+ RUN git checkout v0.12.0
24+ # RUN if [ -n "$VLLM_COMMIT_HASH" ]; then \
25+ # git checkout $VLLM_COMMIT_HASH; \
26+ # fi
2627RUN pip install -r requirements/tpu.txt --retries 3
2728RUN VLLM_TARGET_DEVICE="tpu" pip install -e .
2829
You can’t perform that action at this time.
0 commit comments