diff --git a/README.md b/README.md index f982ce6..5bbf1f9 100644 --- a/README.md +++ b/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. 下载训练底模 - -从网上下载 `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 diff --git a/hair_service_sd/gen_super_image.py b/hair_service_sd/gen_super_image.py index 1d86d9a..b315eca 100644 --- a/hair_service_sd/gen_super_image.py +++ b/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_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": { } diff --git a/photo_service/lora_train_service_1.py b/photo_service/lora_train_service_1.py index 3bec5a8..eff8401 100644 --- a/photo_service/lora_train_service_1.py +++ b/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 ' diff --git a/setup.sh b/setup.sh index ec49ace..f3810b9 100755 --- a/setup.sh +++ b/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..." -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. 创建运行时目录