File tree Expand file tree Collapse file tree
.github/actions/setup-vllm Expand file tree Collapse file tree Original file line number Diff line number Diff line change 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}"
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
Original file line number Diff line number Diff line change 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
77RUN pip install "huggingface-hub[cli]"
88
You can’t perform that action at this time.
0 commit comments