25 lines
772 B
Bash
25 lines
772 B
Bash
#!/usr/bin/env bash
|
|
set -euo pipefail
|
|
|
|
# ---------- 配置 ----------
|
|
HOST_PORT="${HOST_PORT:-8904}"
|
|
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
|
|
|
# ---------- Python 虚拟环境 ----------
|
|
if [ -f "${SCRIPT_DIR}/../myenv/bin/activate" ]; then
|
|
source "${SCRIPT_DIR}/../myenv/bin/activate"
|
|
elif [ -f "${SCRIPT_DIR}/../../myenv/bin/activate" ]; then
|
|
source "${SCRIPT_DIR}/../../myenv/bin/activate"
|
|
fi
|
|
|
|
# ---------- 可选 CUDA 设备隔离 ----------
|
|
CUDA_DEV="${CUDA_VISIBLE_DEVICES:-}"
|
|
if [ -n "${CUDA_DEV}" ]; then
|
|
export CUDA_VISIBLE_DEVICES="${CUDA_DEV}"
|
|
echo "[Z-Image] CUDA_VISIBLE_DEVICES=${CUDA_DEV}"
|
|
fi
|
|
|
|
echo "[Z-Image] 启动图像生成服务 (端口=${HOST_PORT})"
|
|
cd "${SCRIPT_DIR}"
|
|
exec python image_gen_server.py --port "${HOST_PORT}"
|