# 脸部 LoRA 训练脚本（Qwen-Image Edit-2511 / 16G 显存 / 32G 内存优化版）
# 用法：powershell -ExecutionPolicy Bypass -File 训练脚本.ps1
# 前置：素材已整理到 output/train_dataset/（img_001.jpg + img_001.txt 打标）

$ErrorActionPreference = "Stop"
$env:PYTHONUTF8 = "1"

$VENV = "D:\AI\sd\musubi-tuner\.venv\Scripts"
$TOOL = "D:\AI\sd\musubi-tuner\src\musubi_tuner"
$MODELS = "D:\AI\sd\models\qwen-edit-2511"
$PROJ = "D:\F\NewI\opencode\daily-workspace\projects\脸部LoRA训练-Qwen-Image"

$env:HF_HUB_OFFLINE = "1"   # 全程离线，避免联网

& "$VENV\accelerate.exe" launch --num_cpu_threads_per_process 1 --mixed_precision bf16 `
    "$TOOL\qwen_image_train_network.py" `
    --dit "$MODELS\transformer\diffusion_pytorch_model-00001-of-00005.safetensors" `
    --vae "$MODELS\diffusion_pytorch_model.safetensors" `
    --text_encoder "$MODELS\text_encoder\model-00001-of-00004.safetensors" `
    --model_version edit-2511 `
    --dataset_config "$PROJ\config\dataset.toml" `
    --sdpa --mixed_precision bf16 `
    --timestep_sampling shift --weighting_scheme none --discrete_flow_shift 2.2 `
    --optimizer_type adamw8bit --learning_rate 5e-5 `
    --gradient_checkpointing `
    --network_module networks.lora_qwen_image --network_dim 16 `
    --fp8_base --fp8_scaled --fp8_vl --blocks_to_swap 32 `
    --max_train_epochs 80 --save_every_n_epochs 10 --seed 42 `
    --output_dir "$PROJ\output\checkpoints" --output_name myface_lora