colomi
4b959b3a98
- lora_train_service_1.py: 将 sudo docker run 命令替换为本地 python -m accelerate.commands.launch，通过 PYTHONPATH 复用 .local/ 下的 torch/accelerate/bitsandbytes 等包 - setup.sh: 添加 kohya 环境到 conda-pack 恢复列表 - start_all_services.sh: 导出 CONDA_BASE 供 photo_service 读取 - README.md: 添加 kohya.tar.gz 到网盘下载清单 - 已验证: 训练命令可完整执行（加载底模→创建LoRA→训练→保存）
25 lines
963 B
Bash
Executable File
25 lines
963 B
Bash
Executable File
#!/bin/bash
|
|
BASE_DIR="$(cd "$(dirname "$0")" && pwd)"
|
|
CONDA_BASE="${CONDA_BASE:-/home/szlc/miniconda3}"
|
|
export CONDA_BASE
|
|
|
|
mkdir -p "$BASE_DIR/logs"
|
|
|
|
# 1. hair_service_sd (端口 8801)
|
|
cd "$BASE_DIR/hair_service_sd"
|
|
nohup "$CONDA_BASE/envs/my_hair/bin/python" run_copy_cost_colorb64.py > "$BASE_DIR/logs/hair_service.log" 2>&1 &
|
|
echo "hair_service_sd 已启动 (PID: $!, 端口 8801)"
|
|
|
|
# 2. stable-diffusion-webui (端口 57860)
|
|
cd "$BASE_DIR/stable-diffusion-webui"
|
|
nohup "$CONDA_BASE/envs/sdwebui/bin/python" webui.py --api --listen --xformers --port 57860 > "$BASE_DIR/logs/webui.log" 2>&1 &
|
|
echo "stable-diffusion-webui 已启动 (PID: $!, 端口 57860)"
|
|
|
|
# 3. photo_service (端口 32678)
|
|
cd "$BASE_DIR/photo_service"
|
|
nohup "$CONDA_BASE/envs/py310/bin/python" lora_train_service_1.py > "$BASE_DIR/logs/photo_service.log" 2>&1 &
|
|
echo "photo_service 已启动 (PID: $!, 端口 32678)"
|
|
|
|
echo ""
|
|
echo "所有服务已在后台启动，日志在 $BASE_DIR/logs/"
|