包含: - hair_service_sd: 换发型/换发色算法服务 (端口 8801) - photo_service: LoRA 训练调度服务 (端口 32678) - stable-diffusion-webui: SD WebUI 推理服务 (端口 57860) - kohya_ss_home: 训练环境代码 - meidaojia: 监控测试脚本 - setup.sh: 一键部署脚本 (conda环境恢复 + 配置生成 + 完整性检查) - start_all_services.sh: 启动3个服务 - configure.ini.template: 路径模板化 (BASE_DIR自动推导) - conda_envs/py310.yml: py310 环境定义 大文件 (weights/, models/, data/, conda_envs/*.tar.gz 等) 通过 .gitignore 排除, 由网盘单独上传。
28 lines
946 B
Markdown
28 lines
946 B
Markdown
换发算法服务代码
|
||
|
||
相关配置:config/configure.ini \
|
||
模型:请存放于weights目录下 \
|
||
服务启动:执行 gunicorn server:app -c gunicorn_config.py \
|
||
测试相关代码请写在 unit_test.py里
|
||
|
||
建议环境:
|
||
python3.7,pytorch1.8,cu11.1,torchvision10.0 \
|
||
gunicorn需要在/usr/bin/gunicorn需要在里修改python地址
|
||
|
||
run run_copy_cost.py
|
||
|
||
|
||
|
||
需要启动的服务:
|
||
环境变量设置
|
||
export CRYPTOGRAPHY_OPENSSL_NO_LEGACY=1
|
||
1. 换发算法服务
|
||
cd /root/project/hair_service_sd
|
||
/usr/local/miniconda3/envs/condiff-train-hair/bin/python run_copy_cost_colorb64.py
|
||
2. webUI服务推理 + onediff
|
||
cd /root/project/onediff/stable-diffusion-webui
|
||
/usr/local/miniconda3/envs/onediff/bin/python python webui.py --api --listen --xformers --port 57860
|
||
./webui.sh --api --listen --disable-safe-unpickle --port 9038
|
||
3. photo_service
|
||
cd /root/project/photo_service
|
||
/usr/local/miniconda3/envs/py310/bin/python lora_train_service_1.py |