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

Signed-off-by: Peter Schuurman <psch@google.com>
This commit is contained in:
pwschuurman 2025-10-08 23:48:51 -07:00 committed by GitHub
parent 1b2c440cd6
commit 0d7c3cb51d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -542,7 +542,7 @@ RUN --mount=type=cache,target=/root/.cache/uv \
else \
BITSANDBYTES_VERSION="0.46.1"; \
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