x
This commit is contained in:
+18
-7
@@ -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 "将使用容器内默认配置"
|
||||
|
||||
Reference in New Issue
Block a user