adapt to sglang v0.5.2rc1 on dcu
This commit is contained in:
31
docker/serve
Executable file
31
docker/serve
Executable file
@@ -0,0 +1,31 @@
|
||||
#!/bin/bash
|
||||
|
||||
echo "Starting server"
|
||||
|
||||
SERVER_ARGS="--host 0.0.0.0 --port 8080"
|
||||
|
||||
if [ -n "$TENSOR_PARALLEL_DEGREE" ]; then
|
||||
SERVER_ARGS="${SERVER_ARGS} --tp-size ${TENSOR_PARALLEL_DEGREE}"
|
||||
fi
|
||||
|
||||
if [ -n "$DATA_PARALLEL_DEGREE" ]; then
|
||||
SERVER_ARGS="${SERVER_ARGS} --dp-size ${DATA_PARALLEL_DEGREE}"
|
||||
fi
|
||||
|
||||
if [ -n "$EXPERT_PARALLEL_DEGREE" ]; then
|
||||
SERVER_ARGS="${SERVER_ARGS} --ep-size ${EXPERT_PARALLEL_DEGREE}"
|
||||
fi
|
||||
|
||||
if [ -n "$MEM_FRACTION_STATIC" ]; then
|
||||
SERVER_ARGS="${SERVER_ARGS} --mem-fraction-static ${MEM_FRACTION_STATIC}"
|
||||
fi
|
||||
|
||||
if [ -n "$QUANTIZATION" ]; then
|
||||
SERVER_ARGS="${SERVER_ARGS} --quantization ${QUANTIZATION}"
|
||||
fi
|
||||
|
||||
if [ -n "$CHUNKED_PREFILL_SIZE" ]; then
|
||||
SERVER_ARGS="${SERVER_ARGS} --chunked-prefill-size ${CHUNKED_PREFILL_SIZE}"
|
||||
fi
|
||||
|
||||
python3 -m sglang.launch_server --model-path /opt/ml/model $SERVER_ARGS
|
||||
Reference in New Issue
Block a user