34 lines
1.1 KiB
PowerShell
34 lines
1.1 KiB
PowerShell
# LoRA train script by @Akegarasu
|
|
|
|
$config_file = "./config/default.toml" # config file | 使用 toml 文件指定训练参数
|
|
$sample_prompts = "./config/sample_prompts.txt" # prompt file for sample | 采样 prompts 文件, 留空则不启用采样功能
|
|
|
|
$sdxl = 0 # train sdxl LoRA | 训练 SDXL LoRA
|
|
$multi_gpu = 0 # multi gpu | 多显卡训练 该参数仅限在显卡数 >= 2 使用
|
|
|
|
# ============= DO NOT MODIFY CONTENTS BELOW | 请勿修改下方内容 =====================
|
|
|
|
# Activate python venv
|
|
.\venv\Scripts\activate
|
|
|
|
$Env:HF_HOME = "huggingface"
|
|
$Env:PYTHONUTF8 = 1
|
|
|
|
$ext_args = [System.Collections.ArrayList]::new()
|
|
$launch_args = [System.Collections.ArrayList]::new()
|
|
|
|
if ($multi_gpu) {
|
|
[void]$launch_args.Add("--multi_gpu")
|
|
[void]$launch_args.Add("--num_processes=2")
|
|
}
|
|
|
|
# run train
|
|
$script_name = if ($sdxl) { "sdxl_train_network.py" } else { "train_network.py" }
|
|
python -m accelerate.commands.launch $launch_args --num_cpu_threads_per_process=8 "./scripts/$script_name" `
|
|
--config_file=$config_file `
|
|
--sample_prompts=$sample_prompts `
|
|
$ext_args
|
|
|
|
Write-Output "Train finished"
|
|
Read-Host | Out-Null
|