RemoveAnythingSimplify/start_api_server.sh at main · foocker/RemoveAnythingSimplify · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
#!/bin/bash

# RemoveAnything API Server 启动脚本
# 使用Uvicorn运行FastAPI应用

set -e

echo "========================================="
echo "RemoveAnything API Server"
echo "========================================="

# 工作目录
cd "$(dirname "$0")"
WORK_DIR="$(pwd)"
echo "工作目录: $WORK_DIR"

# 激活conda环境
source /root/miniconda3/etc/profile.d/conda.sh
conda activate nunchaku

echo ""
echo "环境信息:"
echo "  Python: $(python --version)"
echo "  PyTorch: $(python -c 'import torch; print(torch.__version__)')"
echo "  Nunchaku: $(python -c 'import nunchaku; print(nunchaku.__version__)')"
echo ""

# CUDA库路径配置
export LD_LIBRARY_PATH="/root/miniconda3/envs/nunchaku/lib/python3.12/site-packages/torch/lib:/usr/local/cuda-12.4/lib64:${LD_LIBRARY_PATH}"

# 默认参数
HOST="${HOST:-0.0.0.0}"
PORT="${PORT:-8000}"
WORKERS="${WORKERS:-1}"
RELOAD="${RELOAD:-false}"

# 解析命令行参数
while [[ $# -gt 0 ]]; do
    case $1 in
        --host)
            HOST="$2"
            shift 2
            ;;
        --port)
            PORT="$2"
            shift 2
            ;;
        --workers)
            WORKERS="$2"
            shift 2
            ;;
        --reload)
            RELOAD="true"
            shift
            ;;
        *)
            echo "未知参数: $1"
            echo "用法: $0 [--host HOST] [--port PORT] [--workers WORKERS] [--reload]"
            exit 1
            ;;
    esac
done

echo "服务配置:"
echo "  - Host: $HOST"
echo "  - Port: $PORT"
echo "  - Workers: $WORKERS"
echo "  - Auto Reload: $RELOAD"
echo ""

# 检查端口是否被占用
if lsof -Pi :$PORT -sTCP:LISTEN -t >/dev/null 2>&1 ; then
    echo "⚠️  警告: 端口 $PORT 已被占用"
    echo "   正在尝试清理..."
    kill -9 $(lsof -t -i:$PORT) 2>/dev/null || true
    sleep 2
fi

echo "========================================="
echo "启动FastAPI服务..."
echo "========================================="

# 构建uvicorn命令
CMD="uvicorn api_server:app --host $HOST --port $PORT --workers $WORKERS --log-level info"

if [ "$RELOAD" = "true" ]; then
    CMD="$CMD --reload"
fi

echo "执行命令: $CMD"
echo ""
echo "API文档: http://$HOST:$PORT/docs"
echo "按 Ctrl+C 停止服务"
echo "========================================="
echo ""

# 启动服务
$CMD