Skip to content

Commit 67867ef

Browse files
authored
Merge branch 'main' into feat/112-bedrock-live-ci
2 parents 9e6e99b + ea91cd9 commit 67867ef

2 files changed

Lines changed: 12 additions & 2 deletions

File tree

.github/actions/setup-vllm/action.yml

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,8 +17,14 @@ runs:
1717
MODEL_DIR="${MODEL_DIR:-/opt/vllm/models}"
1818
1919
# Set VLLM_ARGS based on VLLM_MODE
20+
# Note: the inference and embedding containers run concurrently on the
21+
# same host, and on the CPU backend --gpu-memory-utilization reserves
22+
# that fraction of *total* host RAM (despite its name). vLLM 0.27.1
23+
# added a pre-flight check that fails startup when the reserved fraction
24+
# exceeds currently-available RAM, so the inference container is kept
25+
# lean (0.30) to leave room for the embedding container to start.
2026
if [[ "$VLLM_MODE" == "inference" ]]; then
21-
VLLM_ARGS="--host 0.0.0.0 --port 8000 --enforce-eager --language-model-only --skip-mm-profiling --enable-auto-tool-choice --tool-call-parser hermes --model ${MODEL_DIR}/Qwen/Qwen3.5-0.8B --served-model-name Qwen/Qwen3.5-0.8B --max-model-len 8192 --gpu-memory-utilization 0.50"
27+
VLLM_ARGS="--host 0.0.0.0 --port 8000 --enforce-eager --language-model-only --skip-mm-profiling --enable-auto-tool-choice --tool-call-parser hermes --model ${MODEL_DIR}/Qwen/Qwen3.5-0.8B --served-model-name Qwen/Qwen3.5-0.8B --max-model-len 8192 --gpu-memory-utilization 0.30"
2228
VLLM_PORT=8000
2329
elif [[ "$VLLM_MODE" == "embedding" ]]; then
2430
VLLM_ARGS="--host 0.0.0.0 --port 8001 --model ${MODEL_DIR}/ibm-granite/granite-embedding-125m-english --served-model-name ibm-granite/granite-embedding-125m-english --gpu-memory-utilization 0.25 --hf-overrides {\"is_matryoshka\":true}"
@@ -29,9 +35,13 @@ runs:
2935
fi
3036
3137
# Start vllm container
38+
# --shm-size: vLLM's multiproc executor allocates its inter-process
39+
# message queue in /dev/shm; the Docker default of 64 MiB is too small
40+
# (v0.27.1 requires 160 MiB) and EngineCore fails to start without this.
3241
docker run -d \
3342
--name vllm-$VLLM_MODE \
3443
--net=host \
44+
--shm-size=2g \
3545
$VLLM_IMAGE \
3646
$VLLM_ARGS
3747

vllm/Containerfile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
#
33
# Reference: https://hub.docker.com/r/vllm/vllm-openai-cpu
44

5-
FROM vllm/vllm-openai-cpu:v0.26.0
5+
FROM vllm/vllm-openai-cpu:v0.27.1
66

77
RUN pip install "huggingface-hub[cli]"
88

0 commit comments

Comments
 (0)