部署修复: - torch.load 增加 weights_only=False patch,兼容 PyTorch 2.6+ 加载旧权重 - OSS 改为懒加载,本地用 output_format=base64 无需配凭证即可启动 - 补全被 gitignore 误排除的必需代码:core/models/layers/data、models/layers/data、keypoints/lib - webui 训练命令 --xformers 改 --sdpa(修复 xformers 无 CUDA 支持报错) 功能调整: - hair_grow_service 端口改 8899、preview 路由修复(send_file) - list_hairstyles 增加发型白名单,测试页只展示当前5个发型 新增脚本: - train_lora_parallel.py:直接调 kohya 并行训练 LoRA(绕过 photo_service 串行限制) - train_hairstyles_parallel.py / train_batch_stepC.py:批量训练辅助脚本 - scripts/sync_data_to_server.sh:大文件断点续传到云服务器 文档: - docs/换发型集成文档.md:换发型完整流程、服务架构、资源依赖、训练方法、集成步骤
换发算法服务代码
相关配置: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
- 换发算法服务 cd /root/project/hair_service_sd /usr/local/miniconda3/envs/condiff-train-hair/bin/python run_copy_cost_colorb64.py
- 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
- photo_service cd /root/project/photo_service /usr/local/miniconda3/envs/py310/bin/python lora_train_service_1.py