asdf

2026-07-17 22:13:56 +08:00
parent 2606aca7c5
commit b2738fba7d
28 changed files with 126 additions and 126 deletions
@@ -1,21 +1,21 @@
 [default]
 modelDir = weights
-hairstyleDir = /home/xsl/change_hair/project/data/ref_hairstyle
-haircolorDir= /home/xsl/change_hair/project/data/ref_haircolor
-userDir=/home/xsl/change_hair/project/data/userImage
-tmp_dir=/home/xsl/change_hair/project/data/tmp
-res_dir=/home/xsl/change_hair/project/data/res_dir
-userInfo_dir=/home/xsl/change_hair/project/data/user_info
+hairstyleDir = /home/ubuntu/change_hair/project/data/ref_hairstyle
+haircolorDir= /home/ubuntu/change_hair/project/data/ref_haircolor
+userDir=/home/ubuntu/change_hair/project/data/userImage
+tmp_dir=/home/ubuntu/change_hair/project/data/tmp
+res_dir=/home/ubuntu/change_hair/project/data/res_dir
+userInfo_dir=/home/ubuntu/change_hair/project/data/user_info
 baseColor_ID=HDR10_443322
 Port = 11023
-refer_dir = /home/xsl/change_hair/project/data/ref_online
-ref_user_dir = /home/xsl/change_hair/project/data/ref_user_imgs
-train_dir = /home/xsl/change_hair/data/train_material
-refImgDir=/home/xsl/change_hair/project/data/refImage
-hair_template_material_dir=/home/xsl/change_hair/project/data/hair_template_material
-ref_color=/home/xsl/change_hair/project/data/ref_color
-ref_color_img=/home/xsl/change_hair/project/data/ref_color_imgs
-upload_train_dir=/home/xsl/change_hair/project/data/upload_train_imgs
+refer_dir = /home/ubuntu/change_hair/project/data/ref_online
+ref_user_dir = /home/ubuntu/change_hair/project/data/ref_user_imgs
+train_dir = /home/ubuntu/change_hair/data/train_material
+refImgDir=/home/ubuntu/change_hair/project/data/refImage
+hair_template_material_dir=/home/ubuntu/change_hair/project/data/hair_template_material
+ref_color=/home/ubuntu/change_hair/project/data/ref_color
+ref_color_img=/home/ubuntu/change_hair/project/data/ref_color_imgs
+upload_train_dir=/home/ubuntu/change_hair/project/data/upload_train_imgs
 ;version=local
 version=online

@@ -23,7 +23,7 @@ version=online
 strength=1

 [logger]
-logpath = /home/xsl/change_hair/project/logs
+logpath = /home/ubuntu/change_hair/project/logs
 level=INFO

 [timelogger]
@@ -25,7 +25,7 @@ import cv2
 import numpy as np

 # MediaPipe 模型路径
-MP_MODEL = "/home/xsl/change_hair/project/hair_service_sd/weights/mediapipe/face_landmarker.task"
+MP_MODEL = "/home/ubuntu/change_hair/project/hair_service_sd/weights/mediapipe/face_landmarker.task"
 _mp_detector = None


@@ -4,7 +4,7 @@
 export CRYPTOGRAPHY_OPENSSL_NO_LEGACY=1

 # 创建Lora目录软连接
-TARGET_LORA="/home/xsl/change_hair/project/onediff/stable-diffusion-webui/models/Lora"
+TARGET_LORA="/home/ubuntu/change_hair/project/onediff/stable-diffusion-webui/models/Lora"
 if [ -e "$TARGET_LORA" ]; then
    echo "删除已存在的Lora连接或目录: $TARGET_LORA"
    rm -rf "$TARGET_LORA"
@@ -13,16 +13,16 @@ ln -s /gz-fs/Lora "$TARGET_LORA"
 echo "创建软连接成功: /gz-fs/Lora -> $TARGET_LORA"

 # 1. 换发算法服务
-cd /home/xsl/change_hair/project/hair_service_sd
-nohup /home/xsl/miniconda3/envs/condiff-train-hair/bin/python run_copy_cost_colorb64.py > hair_service.log 2>&1 &
+cd /home/ubuntu/change_hair/project/hair_service_sd
+nohup /home/ubuntu/miniconda3/envs/condiff-train-hair/bin/python run_copy_cost_colorb64.py > hair_service.log 2>&1 &

 # 2. webUI服务推理 + onediff
-cd /home/xsl/change_hair/project/onediff/stable-diffusion-webui
-nohup /home/xsl/miniconda3/envs/onediff/bin/python webui.py --api --listen --xformers --port 57860 > webui.log 2>&1 &
+cd /home/ubuntu/change_hair/project/onediff/stable-diffusion-webui
+nohup /home/ubuntu/miniconda3/envs/onediff/bin/python webui.py --api --listen --xformers --port 57860 > webui.log 2>&1 &
 nohup ./webui.sh --api --listen --disable-safe-unpickle --port 9038 > webui_sh.log 2>&1 &

 # 3. photo_service
-cd /home/xsl/change_hair/project/photo_service
-nohup /home/xsl/miniconda3/envs/py310/bin/python lora_train_service_1.py > photo_service.log 2>&1 &
+cd /home/ubuntu/change_hair/project/photo_service
+nohup /home/ubuntu/miniconda3/envs/py310/bin/python lora_train_service_1.py > photo_service.log 2>&1 &

 echo "所有服务已在后台启动"
@@ -37,8 +37,8 @@ if version == "local":
    callback_url = 'http://service.aicloud.fit:7395/api/hair/trainCallBack'
 else:
    current_url = 'http://0.0.0.0:7393/'
-    kohya_ss_home_dir = '/home/xsl/change_hair/project/kohya_ss_home'
-    webui_lora_dir = '/home/xsl/change_hair/project/onediff/stable-diffusion-webui/models/Lora'
+    kohya_ss_home_dir = '/home/ubuntu/change_hair/project/kohya_ss_home'
+    webui_lora_dir = '/home/ubuntu/change_hair/project/onediff/stable-diffusion-webui/models/Lora'
    inference_use_onediff = False
    callback_url = 'http://0.0.0.0:8801/api/hair/trainCallBack'
 base_webui_port = '57860'
@@ -307,13 +307,13 @@ def train_thread(sq, gpu_id):
            #   3. GPU 固定 device=0(单卡)
            #   4. 去掉 tokenizer_cache_dir(改用 HF 本地缓存 + 离线模式)
            #   5. 设置 HF_HUB_OFFLINE 避免联网检查
-            kohya_python = '/home/xsl/miniconda3/envs/kohya/bin/python'
+            kohya_python = '/home/ubuntu/miniconda3/envs/kohya/bin/python'
            kohya_workdir = os.path.join(kohya_ss_home_dir, 'kohya_ss')
-            base_model = '/home/xsl/change_hair/project/onediff/stable-diffusion-webui/models/Stable-diffusion/v1-5-pruned-emaonly.safetensors'
+            base_model = '/home/ubuntu/change_hair/project/onediff/stable-diffusion-webui/models/Stable-diffusion/v1-5-pruned-emaonly.safetensors'
            cmd_train = (
                f'cd {kohya_workdir} && '
                f'HF_HUB_OFFLINE=1 TRANSFORMERS_OFFLINE=1 CUDA_VISIBLE_DEVICES={device_id} '
-                f'/home/xsl/miniconda3/envs/kohya/bin/accelerate launch --num_cpu_threads_per_process=2 "./train_network.py" --enable_bucket '
+                f'/home/ubuntu/miniconda3/envs/kohya/bin/accelerate launch --num_cpu_threads_per_process=2 "./train_network.py" --enable_bucket '
                f'--min_bucket_reso=256 --max_bucket_reso=2048 --pretrained_model_name_or_path="{base_model}" '
                f'--train_data_dir={images_dir} --resolution="2000,2000" '
                f'--output_dir={model_dir} '