File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -44,6 +44,8 @@ print_iter_log: true
4444stream_interval: 20
4545EOF
4646
47+ SERVER_LOG=$( mktemp /tmp/server-XXXXXX.log)
48+
4749# mpirun -n 1 --oversubscribe --allow-run-as-root trtllm-serve $MODEL --tp_size $TP --trust_remote_code --max_seq_len $MAX_MODEL_LEN --max_num_tokens $MAX_MODEL_LEN --num_postprocess_workers 2 --extra_llm_api_options llama-config.yml --port $PORT > $SERVER_LOG 2>&1 &
4850mpirun -n 1 --oversubscribe --allow-run-as-root \
4951trtllm-serve $MODEL \
@@ -58,14 +60,18 @@ trtllm-serve $MODEL \
5860--port $PORT \
5961--tp_size=$TP \
6062--pp_size=1 \
61- 2>&1 | tee $( mktemp /tmp/server-XXXXXX.log) &
63+ > $SERVER_LOG 2>&1 &
64+
65+ # Show logs until server is ready
66+ tail -f $SERVER_LOG &
67+ TAIL_PID=$!
6268
6369# Show server logs til' it is up, then stop showing
6470set +x
6571until curl --output /dev/null --silent --fail http://0.0.0.0:$PORT /health; do
6672 sleep 5
6773done
68- pkill -P $$ tee 2> /dev/null
74+ kill $TAIL_PID
6975
7076set -x
7177BENCH_SERVING_DIR=$( mktemp -d /tmp/bmk-XXXXXX)
You can’t perform that action at this time.
0 commit comments