This commit is contained in:
2026-03-10 23:57:37 +08:00
parent a48423d817
commit 65ccd278c9
4 changed files with 63 additions and 312 deletions
+18 -7
View File
@@ -5,15 +5,26 @@
set -e
# 配置变量
# ============== 配置变量(可修改) ==============
# 项目目录
PROJECT_DIR="/opt/project/vllm-r9700-container"
# 挂载配置
MODEL_DIR="${MODEL_DIR:-/opt/model}" # 本地模型目录(独立于项目目录)
CONFIG_FILE="${CONFIG_FILE:-${PROJECT_DIR}/model_config.yaml}" # 本地配置文件(项目根目录下)
# 容器内配置
LOCAL_MODEL_DIR="${LOCAL_MODEL_DIR:-/model}" # 容器内模型目录(与 Dockerfile 保持一致)
CONFIG_FILE_IN_CONTAINER="/etc/vllm/model_config.yaml" # 容器内配置文件路径
# Docker 镜像配置
IMAGE_NAME="custom-vllm-r9700"
IMAGE_TAG="latest"
CONTAINER_NAME="custom-vllm-r9700"
PROJECT_DIR="/opt/project/vllm-r9700-container"
MODEL_DIR="${MODEL_DIR:-/opt/models}"
CONFIG_FILE="${CONFIG_FILE:-/opt/model_config.yaml}"
LOCAL_MODEL_DIR="${LOCAL_MODEL_DIR:-/models}"
# Git 配置
GIT_BRANCH="${GIT_BRANCH:-main}"
# ================================================
# 颜色定义
RED='\033[0;31m'
@@ -286,8 +297,8 @@ run_container() {
fi
if [ -f "$CONFIG_FILE" ]; then
VOLUME_ARGS="$VOLUME_ARGS -v ${CONFIG_FILE}:/etc/vllm/model_config.yaml"
print_info "配置文件:${CONFIG_FILE}"
VOLUME_ARGS="$VOLUME_ARGS -v ${CONFIG_FILE}:${CONFIG_FILE_IN_CONTAINER}"
print_info "配置文件:${CONFIG_FILE} -> ${CONFIG_FILE_IN_CONTAINER}"
else
print_warning "配置文件不存在:${CONFIG_FILE}"
print_info "将使用容器内默认配置"