Skip to content

Commit c9c1b31

Browse files
committed
fix server issue
1 parent 95657d9 commit c9c1b31

1 file changed

Lines changed: 3 additions & 4 deletions

File tree

.github/scripts/run_vllm_profiling.sh

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -47,9 +47,8 @@ wait_for_server() {
4747
kill_gpu_processes() {
4848
echo "Cleaning up processes..."
4949
lsof -t -i:${PORT} | xargs -r kill -9 2>/dev/null || true
50-
pgrep -f "vllm serve" | xargs -r kill -9 2>/dev/null || true
50+
pgrep -f "vllm" | xargs -r kill -9 2>/dev/null || true
5151
pgrep python3 | xargs -r kill -9 2>/dev/null || true
52-
pgrep python | xargs -r kill -9 2>/dev/null || true
5352

5453
# Wait until GPU memory usage decreases
5554
if command -v nvidia-smi; then
@@ -65,9 +64,9 @@ kill_gpu_processes
6564

6665
# Start vLLM server in the background
6766
echo "Starting vLLM server..."
68-
echo "Server command: VLLM_USE_V1=${VLLM_USE_V1} vllm serve ${MODEL_NAME} --swap-space 16 --disable-log-requests --host :: --port ${PORT} --dtype float16"
6967

70-
VLLM_USE_V1=${VLLM_USE_V1} vllm serve "${MODEL_NAME}" \
68+
VLLM_USE_V1=${VLLM_USE_V1} python3 -m vllm.entrypoints.openai.api_server \
69+
--model "${MODEL_NAME}" \
7170
--swap-space 16 \
7271
--disable-log-requests \
7372
--host :: \

0 commit comments

Comments
 (0)