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

- 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 下载步骤

README.md

@@ -50,14 +50,7 @@ cd ~/change_hair_3090
 | conda_envs/my_hair.tar.gz | ~5G | conda 环境 |
 | conda_envs/sdwebui.tar.gz | ~4G | conda 环境 |
 
-### 3. 下载训练底模
+### 3. 运行部署脚本
-
-从网上下载 `majicmixRealistic_v7.safetensors`，放置到：
-```
-stable-diffusion-webui/models/Stable-diffusion/majicmixRealistic_v7.safetensors
-```
-
-### 4. 运行部署脚本
 
 ```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

@@ -182,8 +182,8 @@ class ControlnetRequestImg2Img:
             "inpainting_fill": 1,
             "inpainting_mask_invert": 0,
             "mask": self.read_mask(),
-            "refiner_checkpoint": "majicmixRealistic_v7.safetensors",
+            # "refiner_checkpoint": "majicmixRealistic_v7.safetensors",  # 底模不存在，禁用 refiner
-            "refiner_switch_at": 0.5,
+            # "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_checkpoint": "majicmixRealistic_v7.safetensors",  # 底模不存在，禁用 refiner
-            "refiner_switch_at": 0.5,
+            # "refiner_switch_at": 0.5,
             "denoising_strength": 0.7,
             "alwayson_scripts": {
             }

photo_service/lora_train_service_1.py

@@ -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

@@ -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..."
+echo "[6/7] 检查训练底模 v1-5-pruned-emaonly..."
-if [ -f "$BASE_DIR/stable-diffusion-webui/models/Stable-diffusion/majicmixRealistic_v7.safetensors" ]; then
+if [ -f "$BASE_DIR/stable-diffusion-webui/models/Stable-diffusion/v1-5-pruned-emaonly.safetensors" ]; then
-    echo "  ✓ majicmixRealistic_v7.safetensors 已存在"
+    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. 创建运行时目录