initial open-source release
This commit is contained in:
commit
01c4605498
224 files changed
+55124
No files matched your search
@@ -0,0 +1,47 @@
|
||||
#!/usr/bin/env bash
|
||||
set -euo pipefail
|
||||
|
||||
# 自动获取脚本所在目录
|
||||
SCRIPT_DIR="$(cd -- "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
||||
VENV_DIR="${SCRIPT_DIR}/myenv"
|
||||
MODEL_DIR="${SCRIPT_DIR}/models/Qwen/Qwen3-ASR-1.7B"
|
||||
HOST_PORT="${HOST_PORT:-8901}"
|
||||
|
||||
# 激活虚拟环境
|
||||
source "${VENV_DIR}/bin/activate"
|
||||
|
||||
# 检查端口占用
|
||||
if lsof -i "tcp:${HOST_PORT}" -t >/dev/null 2>&1; then
|
||||
echo "Port ${HOST_PORT} already in use" >&2
|
||||
exit 1
|
||||
fi
|
||||
|
||||
echo "Starting vLLM server on port ${HOST_PORT} ..."
|
||||
echo "Model path: ${MODEL_DIR}"
|
||||
|
||||
# 进入脚本目录
|
||||
cd "${SCRIPT_DIR}"
|
||||
|
||||
# 设置环境变量
|
||||
export VLLM_USE_MODELSCOPE=true
|
||||
export MODELSCOPE_DOWNLOAD_MODE=local
|
||||
export MODELSCOPE_HUB_ENABLE_FILE_CHECK=false
|
||||
export MODELSCOPE_CACHE="${SCRIPT_DIR}/models"
|
||||
export HF_HUB_OFFLINE=1
|
||||
export TRANSFORMERS_OFFLINE=1
|
||||
|
||||
# 固定使用 GPU 0(多 GPU 服务器上所有模型共用一张卡)
|
||||
export CUDA_VISIBLE_DEVICES=3
|
||||
|
||||
# 启动 vLLM 服务
|
||||
vllm serve "${MODEL_DIR}" \
|
||||
--port "${HOST_PORT}" \
|
||||
--host 0.0.0.0 \
|
||||
--quantization fp8 \
|
||||
--dtype auto \
|
||||
--kv-cache-dtype fp8 \
|
||||
--max-model-len 16384 \
|
||||
--allowed-local-media-path / \
|
||||
--tensor-parallel-size 1 \
|
||||
--gpu-memory-utilization 0.15 \
|
||||
--max-num-batched-tokens 4096
|
||||
Reference in new issue
Block a user