initial setup: ComfyUI + kohya_ss scripts, LoRA config, workflows
This commit is contained in:
23
train_lora.ps1
Normal file
23
train_lora.ps1
Normal file
@@ -0,0 +1,23 @@
|
||||
#!/usr/bin/env pwsh
|
||||
# train_lora.ps1 — run LoRA training from a config file
|
||||
# usage: .\train_lora.ps1 training/my_lora.toml
|
||||
|
||||
param(
|
||||
[Parameter(Mandatory=$true)]
|
||||
[string]$config_file
|
||||
)
|
||||
|
||||
$root = $PSScriptRoot
|
||||
|
||||
if (-not (Test-Path "$root/$config_file")) {
|
||||
Write-Host "config file not found: $config_file" -ForegroundColor Red
|
||||
exit 1
|
||||
}
|
||||
|
||||
Set-Location "$root/kohya_ss"
|
||||
.\venv\Scripts\Activate.ps1
|
||||
|
||||
accelerate launch `
|
||||
--num_cpu_threads_per_process 2 `
|
||||
train_network.py `
|
||||
--config_file "$root/$config_file"
|
||||
Reference in New Issue
Block a user