xsl/change_hair_3090
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

移除 majicmixRealistic_v7 依赖，改用 v1-5-pruned-emaonly

Browse Source 
- lora_train_service_1.py: 训练底模改为 v1-5-pruned-emaonly.safetensors
- gen_super_image.py: 注释掉 refiner_checkpoint（majicmix 不存在会导致推理报错）
- setup.sh: 检查项改为 v1-5-pruned-emaonly.safetensors
- README.md: 移除 majicmixRealistic_v7 下载步骤
This commit is contained in:
colomi
2026-07-11 18:26:51 +08:00
parent 0eb61f3e60
commit 40a512a6c4
4 changed files with 11 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files
README.md
+2 -9
View File
@@ -50,14 +50,7 @@ cd ~/change_hair_3090
| conda_envs/my_hair.tar.gz | ~5G | conda 环境 |
| conda_envs/sdwebui.tar.gz | ~4G | conda 环境 |
### 3. 下载训练底模
从网上下载 `majicmixRealistic_v7.safetensors`，放置到：
```
stable-diffusion-webui/models/Stable-diffusion/majicmixRealistic_v7.safetensors
```
### 4. 运行部署脚本
### 3. 运行部署脚本
```bash
chmod +x setup.sh start_all_services.sh
@@ -70,7 +63,7 @@ setup.sh 会：
3. 检查模型和数据目录完整性
4. 创建运行时目录
### 5. 启动服务
### 4. 启动服务
```bash
./start_all_services.sh
hair_service_sd/gen_super_image.py
+4 -4
View File
@@ -182,8 +182,8 @@ class ControlnetRequestImg2Img:
"inpainting_fill": 1,
"inpainting_mask_invert": 0,
"mask": self.read_mask(),
"refiner_checkpoint": "majicmixRealistic_v7.safetensors",
"refiner_switch_at": 0.5,
# "refiner_checkpoint": "majicmixRealistic_v7.safetensors", # 底模不存在，禁用 refiner
# "refiner_switch_at": 0.5,
"denoising_strength": denoising_strength,
"alwayson_scripts": {
}
@@ -210,8 +210,8 @@ class ControlnetRequestImg2Img:
"inpainting_fill": 1,
"inpainting_mask_invert": 0,
# "mask": self.read_mask(),
"refiner_checkpoint": "majicmixRealistic_v7.safetensors",
"refiner_switch_at": 0.5,
# "refiner_checkpoint": "majicmixRealistic_v7.safetensors", # 底模不存在，禁用 refiner
# "refiner_switch_at": 0.5,
"denoising_strength": 0.7,
"alwayson_scripts": {
}
photo_service/lora_train_service_1.py
+1 -1
View File
@@ -309,7 +309,7 @@ def train_thread(sq, gpu_id):
f'-v {sd_models_dir}:/mnt/sd_models '
'-e PATH=/home/chinatszrn/.local/bin -w /home/chinatszrn/kohya_ss '
'--net=host chinatszrn/ubuntu:kohya_ss accelerate launch --num_cpu_threads_per_process=2 "./train_network.py" --enable_bucket '
'--min_bucket_reso=256 --max_bucket_reso=2048 --pretrained_model_name_or_path="/mnt/sd_models/majicmixRealistic_v7.safetensors" '
'--min_bucket_reso=256 --max_bucket_reso=2048 --pretrained_model_name_or_path="/mnt/sd_models/v1-5-pruned-emaonly.safetensors" '
f'--train_data_dir={container_images_dir} --resolution="2000,2000" '
f'--output_dir={container_model_dir} '
'--network_alpha="64" --save_model_as=safetensors --network_module=networks.lora --text_encoder_lr=5e-05 '
setup.sh
+4 -4
View File
@@ -62,11 +62,11 @@ for dir in hair_service_sd/weights stable-diffusion-webui/models/Lora stable-dif
done
# 6. 检查训练底模
echo "[6/7] 检查 majicmixRealistic_v7..."
if [ -f "$BASE_DIR/stable-diffusion-webui/models/Stable-diffusion/majicmixRealistic_v7.safetensors" ]; then
echo " ✓ majicmixRealistic_v7.safetensors 已存在"
echo "[6/7] 检查训练底模 v1-5-pruned-emaonly..."
if [ -f "$BASE_DIR/stable-diffusion-webui/models/Stable-diffusion/v1-5-pruned-emaonly.safetensors" ]; then
echo " ✓ v1-5-pruned-emaonly.safetensors 已存在"
else
echo " ✗ majicmixRealistic_v7.safetensors 缺失，请从网上下载并放置到 stable-diffusion-webui/models/Stable-diffusion/"
echo " ✗ v1-5-pruned-emaonly.safetensors 缺失，请从网盘下载 models/ 目录"
fi
# 7. 创建运行时目录