Skip to content

Commit 87bb9ce

Browse files
youkaichaorasmith
authored andcommitted
[ci/build] sync default value for wheel size (vllm-project#12398)
Signed-off-by: youkaichao <[email protected]>
1 parent a7bf45b commit 87bb9ce

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

.buildkite/check-wheel-size.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@
55
# Read the VLLM_MAX_SIZE_MB environment variable, defaulting to 300 MiB
66
# Note that we have 400 MiB quota, please use it wisely.
77
# See https://github.com/pypi/support/issues/3792 .
8+
# Please also sync the value with the one in Dockerfile.
89
VLLM_MAX_SIZE_MB = int(os.environ.get('VLLM_MAX_SIZE_MB', 300))
910

1011

Dockerfile

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -126,8 +126,8 @@ RUN --mount=type=cache,target=/root/.cache/ccache \
126126

127127
# Check the size of the wheel if RUN_WHEEL_CHECK is true
128128
COPY .buildkite/check-wheel-size.py check-wheel-size.py
129-
# Default max size of the wheel is 250MB
130-
ARG VLLM_MAX_SIZE_MB=250
129+
# sync the default value with .buildkite/check-wheel-size.py
130+
ARG VLLM_MAX_SIZE_MB=300
131131
ENV VLLM_MAX_SIZE_MB=$VLLM_MAX_SIZE_MB
132132
ARG RUN_WHEEL_CHECK=true
133133
RUN if [ "$RUN_WHEEL_CHECK" = "true" ]; then \

0 commit comments

Comments
 (0)