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

chore: 适配本机 xsl 路径（RTX 5090 机器）

Browse Source 
将所有绝对路径从 /home/ubuntu 改回 /home/xsl，覆盖启动脚本、
configure.ini 及训练/测试脚本。纯路径替换，无功能性改动。

本分支(xsl5090)用于本机(RTX 5090)运行；master 维持 /home/ubuntu
供远程 ubuntu 机器使用。
This commit is contained in:
xsl
2026-07-18 16:26:16 +08:00
parent b2738fba7d
commit 9b7324cd26
31 changed files with 162 additions and 162 deletions
Download Patch File Download Diff File Expand all files Collapse all files
train_lora_parallel.py
+6 -6
View File
@@ -6,7 +6,7 @@
通过 ThreadPoolExecutor 控制并发数。
用法:
python /home/ubuntu/change_hair/train_lora_parallel.py
python /home/xsl/change_hair/train_lora_parallel.py
前提：step1 训练数据已准备好（data/train_material/<hid>/images/1_hairstyle/*.png）
"""
@@ -16,11 +16,11 @@ import time
import subprocess
from concurrent.futures import ThreadPoolExecutor, as_completed
TRAIN_DIR = "/home/ubuntu/change_hair/data/train_material"
KOHYA_WORKDIR = "/home/ubuntu/change_hair/project/kohya_ss_home/kohya_ss"
BASE_MODEL = "/home/ubuntu/change_hair/project/onediff/stable-diffusion-webui/models/Stable-diffusion/v1-5-pruned-emaonly.safetensors"
KOHYA_ACCEL = "/home/ubuntu/miniconda3/envs/kohya/bin/accelerate"
LOG_DIR = "/home/ubuntu/change_hair/project/logs"
TRAIN_DIR = "/home/xsl/change_hair/data/train_material"
KOHYA_WORKDIR = "/home/xsl/change_hair/project/kohya_ss_home/kohya_ss"
BASE_MODEL = "/home/xsl/change_hair/project/onediff/stable-diffusion-webui/models/Stable-diffusion/v1-5-pruned-emaonly.safetensors"
KOHYA_ACCEL = "/home/xsl/miniconda3/envs/kohya/bin/accelerate"
LOG_DIR = "/home/xsl/change_hair/project/logs"
# 本次要训练的发型
HAIRSTYLES = ["chang_tuoyuan", "chang_zhixian", "chang_huaban", "chang_xinxing"]