Skip to content

GitLab

Unverified Commit 0d7c3cb5 authored by pwschuurman's avatar pwschuurman Committed by GitHub
Browse files

Update Dockerfile and install runai-model-streamer[gcs] package (#26464) 


Signed-off-by: default avatarPeter Schuurman <psch@google.com>
parent 1b2c440c
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
docker/Dockerfile
View file @ 0d7c3cb5
...@@ -542,7 +542,7 @@ RUN --mount=type=cache,target=/root/.cache/uv \ ...@@ -542,7 +542,7 @@ RUN --mount=type=cache,target=/root/.cache/uv \
else \ else \
BITSANDBYTES_VERSION="0.46.1"; \ BITSANDBYTES_VERSION="0.46.1"; \
fi; \ fi; \
uv pip install --system accelerate hf_transfer modelscope "bitsandbytes>=${BITSANDBYTES_VERSION}" 'timm>=1.0.17' 'runai-model-streamer[s3]>=0.14.0' uv pip install --system accelerate hf_transfer modelscope "bitsandbytes>=${BITSANDBYTES_VERSION}" 'timm>=1.0.17' 'runai-model-streamer[s3,gcs]>=0.14.0'
ENV VLLM_USAGE_SOURCE production-docker-image ENV VLLM_USAGE_SOURCE production-docker-image
......
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