diff --git a/.buildkite/check-wheel-size.py b/.buildkite/check-wheel-size.py index 105d0b9e77a4..e29eb78a9f94 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 261f5440aee4..cb9cf0da5be6 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 \