Skip to content

Commit a1cc6a4

Browse files
mgoinZhathw
authored andcommitted
[CI] Raise VLLM_MAX_SIZE_MB to 500 due to failing Build wheel - CUDA 12.9 (vllm-project#26722)
Signed-off-by: mgoin <mgoin64@gmail.com>
1 parent 3a10648 commit a1cc6a4

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

.buildkite/check-wheel-size.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,11 +5,11 @@
55
import sys
66
import zipfile
77

8-
# Read the VLLM_MAX_SIZE_MB environment variable, defaulting to 450 MiB
8+
# Read the VLLM_MAX_SIZE_MB environment variable, defaulting to 500 MiB
99
# Note that we have 800 MiB quota, please use it wisely.
1010
# See https://github.com/pypi/support/issues/6326 .
1111
# Please also sync the value with the one in Dockerfile.
12-
VLLM_MAX_SIZE_MB = int(os.environ.get("VLLM_MAX_SIZE_MB", 450))
12+
VLLM_MAX_SIZE_MB = int(os.environ.get("VLLM_MAX_SIZE_MB", 500))
1313

1414

1515
def print_top_10_largest_files(zip_file):

docker/Dockerfile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -229,7 +229,7 @@ RUN --mount=type=cache,target=/root/.cache/ccache \
229229
# Check the size of the wheel if RUN_WHEEL_CHECK is true
230230
COPY .buildkite/check-wheel-size.py check-wheel-size.py
231231
# sync the default value with .buildkite/check-wheel-size.py
232-
ARG VLLM_MAX_SIZE_MB=450
232+
ARG VLLM_MAX_SIZE_MB=500
233233
ENV VLLM_MAX_SIZE_MB=$VLLM_MAX_SIZE_MB
234234
ARG RUN_WHEEL_CHECK=true
235235
RUN if [ "$RUN_WHEEL_CHECK" = "true" ]; then \

0 commit comments

Comments
 (0)