#!/usr/bin/env bash
set -euo pipefail

# 自动获取脚本所在目录
SCRIPT_DIR="$(cd -- "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
VENV_DIR="${SCRIPT_DIR}/.venv_audio"
MODEL_DIR="${SCRIPT_DIR}/modelscope/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}/modelscope"
export HF_HUB_OFFLINE=1
export TRANSFORMERS_OFFLINE=1

# 启动 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.10 \
  --max-num-batched-tokens 4096