Skip to content

GitLab

Unverified Commit aea6ad62 authored by Michael Goin's avatar Michael Goin Committed by GitHub
Browse files

Add hf_transfer to testing image (#10096)

parent da07a9ea
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 0 deletions
Dockerfile
View file @ aea6ad62
...@@ -191,6 +191,11 @@ ADD . /vllm-workspace/ ...@@ -191,6 +191,11 @@ ADD . /vllm-workspace/
RUN --mount=type=cache,target=/root/.cache/pip \ RUN --mount=type=cache,target=/root/.cache/pip \
python3 -m pip install -r requirements-dev.txt python3 -m pip install -r requirements-dev.txt
# enable fast downloads from hf (for testing)
RUN --mount=type=cache,target=/root/.cache/pip \
python3 -m pip install hf_transfer
ENV HF_HUB_ENABLE_HF_TRANSFER 1
# Copy in the v1 package for testing (it isn't distributed yet) # Copy in the v1 package for testing (it isn't distributed yet)
COPY vllm/v1 /usr/local/lib/python3.12/dist-packages/vllm/v1 COPY vllm/v1 /usr/local/lib/python3.12/dist-packages/vllm/v1
......
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