mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2026-04-02 17:07:08 +08:00
Merge ff00ef309dc619752c4c22028583e36a6f7e37a1 into 254f6b986720c92ddf97fbb1a6a6465da8e87e29
This commit is contained in:
commit
333b646717
@ -51,6 +51,8 @@ FROM fetch_vllm_${REMOTE_VLLM} AS fetch_vllm
|
||||
# -----------------------
|
||||
# vLLM build stages
|
||||
FROM fetch_vllm AS build_vllm
|
||||
# Set target device to ROCm for build
|
||||
ENV VLLM_TARGET_DEVICE=rocm
|
||||
# Build vLLM
|
||||
RUN cd vllm \
|
||||
&& python3 -m pip install -r requirements/rocm.txt \
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user