Skip to content

GitLab

"...git@developer.sourcefind.cn:tsoc/superbenchmark.git" did not exist on "7f607e4f745b84fdac1c1b693b32bb65ca8a3c79"
Commit 7713e25c authored by Anant Sharma's avatar Anant Sharma Committed by GitHub
Browse files

build: use wheel for vllm install (#163)

parent 0f4529e9
Show whitespace changes
Inline Side-by-side
Showing with 7 additions and 1 deletion
container/Dockerfile.vllm
View file @ 7713e25c
...@@ -170,7 +170,13 @@ RUN --mount=type=bind,source=./container/deps/requirements.txt,target=/tmp/requi ...@@ -170,7 +170,13 @@ RUN --mount=type=bind,source=./container/deps/requirements.txt,target=/tmp/requi
ARG VLLM_REF="v0.7.2" ARG VLLM_REF="v0.7.2"
ARG VLLM_PATCH="vllm_${VLLM_REF}-dynamo-kv-disagg-patch.patch" ARG VLLM_PATCH="vllm_${VLLM_REF}-dynamo-kv-disagg-patch.patch"
RUN --mount=type=bind,source=./container/deps/,target=/tmp/deps \ RUN --mount=type=bind,source=./container/deps/,target=/tmp/deps \
bash /tmp/deps/vllm/install.sh --patch /tmp/deps/vllm/${VLLM_PATCH} --ref ${VLLM_REF} --install-cmd "uv pip install --editable" --use-precompiled --installation-dir /opt/vllm git clone https://github.com/vllm-project/vllm.git vllm && \
cd vllm && \
git checkout ${VLLM_REF} && \
git apply /tmp/deps/vllm/${VLLM_PATCH} && \
VLLM_USE_PRECOMPILED=1 uv build --wheel --out-dir /workspace/dist && \
uv pip install /workspace/dist/vllm*cp312*.whl && \
cd .. && rm -rf vllm
# Install genai-perf for benchmarking # Install genai-perf for benchmarking
ARG GENAI_PERF_TAG="25d0188713adc47868d6b3f22426375237a90529" ARG GENAI_PERF_TAG="25d0188713adc47868d6b3f22426375237a90529"
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment