From 8dd7f8f963ed351f71e66585278a7245d0d91b25 Mon Sep 17 00:00:00 2001 From: youkaichao Date: Fri, 24 Jan 2025 17:54:29 +0800 Subject: [PATCH] [ci/build] sync default value for wheel size (#12398) Signed-off-by: youkaichao Signed-off-by: Isotr0py <2037008807@qq.com> --- .buildkite/check-wheel-size.py | 1 + Dockerfile | 4 ++-- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/.buildkite/check-wheel-size.py b/.buildkite/check-wheel-size.py index 105d0b9e77a44..e29eb78a9f945 100644 --- a/.buildkite/check-wheel-size.py +++ b/.buildkite/check-wheel-size.py @@ -5,6 +5,7 @@ # Read the VLLM_MAX_SIZE_MB environment variable, defaulting to 300 MiB # Note that we have 400 MiB quota, please use it wisely. # See https://github.com/pypi/support/issues/3792 . +# Please also sync the value with the one in Dockerfile. VLLM_MAX_SIZE_MB = int(os.environ.get('VLLM_MAX_SIZE_MB', 300)) diff --git a/Dockerfile b/Dockerfile index 261f5440aee47..cb9cf0da5be65 100644 --- a/Dockerfile +++ b/Dockerfile @@ -126,8 +126,8 @@ RUN --mount=type=cache,target=/root/.cache/ccache \ # Check the size of the wheel if RUN_WHEEL_CHECK is true COPY .buildkite/check-wheel-size.py check-wheel-size.py -# Default max size of the wheel is 250MB -ARG VLLM_MAX_SIZE_MB=250 +# sync the default value with .buildkite/check-wheel-size.py +ARG VLLM_MAX_SIZE_MB=300 ENV VLLM_MAX_SIZE_MB=$VLLM_MAX_SIZE_MB ARG RUN_WHEEL_CHECK=true RUN if [ "$RUN_WHEEL_CHECK" = "true" ]; then \