参数名称
| 参数值
| 分析
|
pretrained_unet_path
| models/kolors/Kolors/unet/diffusion_pytorch_model.safetensors
| 指定预训练UNet模子的路径
|
pretrained_text_encoder_path
| models/kolors/Kolors/text_encoder
| 指定预训练文本编码器的路径
|
pretrained_fp16_vae_path
| models/sdxl-vae-fp16-fix/diffusion_pytorch_model.safetensors
| 指定预训练VAE模子的路径
|
lora_rank
| 16
| 设置LoRA的秩(rank),影响模子的复杂度和性能
|
lora_alpha
| 4
| 设置LoRA的alpha值,控制微调的强度
|
dataset_path
| data/lora_dataset_processed
| 指定用于训练的数据集路径
|
output_path
| ./models
| 指定训练完成后生存模子的路径
|
max_epochs
| 1
| 设置最大训练轮数为1
|
center_crop
|
| 启用中心裁剪,用于图像预处理
|
use_gradient_checkpointing
|
| 启用梯度检查点,节省显存
|
precision
| "16-mixed"
| 设置训练时的精度为混淆16位精度(half precision)
|