qwenasr-qwen36-voxcmp2 base model. -2026/05/14 night

This commit is contained in:
minorQAQ committed 2026-05-15 16:05:32 +08:00
commit 110fe78016
80 files changed
+6057

No files matched your search

+62
View File
@@ -0,0 +1,62 @@
#!/usr/bin/env bash
set -euo pipefail
# 配置区域(可根据实际情况修改)
MODEL_BASE_PATH="${MODEL_BASE_PATH:-/mnt/Qwen3-Omni-30B-A3B-Instruct-NVFP4}"
THINKER_PATH="${MODEL_BASE_PATH}/thinker"
CONTAINER_NAME="${CONTAINER_NAME:-vllm-thinker}"
HOST_PORT="${HOST_PORT:-8091}"
GPU_DEVICES="${GPU_DEVICES:-all}" # 例如 "0,1" 或 "all"
TENSOR_PARALLEL_SIZE="${TENSOR_PARALLEL_SIZE:-1}"
MAX_MODEL_LEN="${MAX_MODEL_LEN:-32768}"
VLLM_IMAGE="${VLLM_IMAGE:-vllm/vllm-openai:latest}"
# 检查模型路径是否存在
if [[ ! -d "$THINKER_PATH" ]]; then
echo "错误: 模型路径不存在: $THINKER_PATH"
echo "请设置环境变量 MODEL_BASE_PATH 指向正确的模型根目录(包含 thinker 子目录)"
exit 1
fi
# 检查 Docker 可用性
if command -v docker >/dev/null 2>&1 && docker info >/dev/null 2>&1; then
DOCKER_CMD="docker"
else
DOCKER_CMD="sudo docker"
fi
# 停止并移除可能已存在的同名容器
${DOCKER_CMD} rm -f "${CONTAINER_NAME}" >/dev/null 2>&1 || true
echo "启动 Thinker 服务..."
echo "模型路径: $THINKER_PATH"
echo "服务端口: $HOST_PORT"
echo "Tensor Parallel Size: $TENSOR_PARALLEL_SIZE"
# 启动容器
${DOCKER_CMD} run -d \
--name "${CONTAINER_NAME}" \
--gpus "${GPU_DEVICES}" \
--ipc=host \
--network host \
-v "${THINKER_PATH}:/model:ro" \
-e VLLM_USE_MODELSCOPE=false \
"${VLLM_IMAGE}" \
sh -c "pip install --no-cache-dir nvidia-modelopt && \
vllm serve /model \
--port ${HOST_PORT} \
--quantization modelopt_fp4 \
--dtype bfloat16 \
--max-model-len ${MAX_MODEL_LEN} \
--tensor-parallel-size ${TENSOR_PARALLEL_SIZE} \
--trust-remote-code \
--enforce-eager"
# 等待服务启动
sleep 3
# 健康检查
echo "健康检查 (等待几秒后):"
echo " curl http://127.0.0.1:${HOST_PORT}/health"
echo "查看日志:"
echo " ${DOCKER_CMD} logs -f ${CONTAINER_NAME}"