richaball style

最初で最後の修正。

#!/bin/bash

# 1. 加载 Conda 环境初始化（AutoDL 默认路径）
source /root/miniconda3/etc/profile.d/conda.sh

# 2. 激活你的虚拟环境
conda activate aki

# 3. 进入训练器目录
cd /root/autodl-tmp/kohya_ss

# 4. 启动训练
accelerate launch \
  --mixed_precision="bf16" \
  ./sd-scripts/sdxl_train_network.py \
  --pretrained_model_name_or_path="/root/autodl-tmp/animagineXL40_v4Zero.safetensors" \
  --train_data_dir="/root/autodl-tmp/richaball/" \
  --network_train_unet_only \
  --resolution="1024,1024" \
  --enable_bucket \
  --min_bucket_reso=512 \
  --max_bucket_reso=2048 \
  --bucket_reso_steps=64 \
  --bucket_no_upscale \
  --network_module="networks.lora" \
  --network_dim=24 \
  --network_alpha=24 \
  --optimizer_type="Prodigy" \
  --learning_rate=1.0 \
  --unet_lr=1.0 \
  --text_encoder_lr=0 \
  --lr_scheduler="cosine" \
  --lr_warmup_steps=200 \
  --optimizer_args "decouple=True" "weight_decay=0.1" "d_coef=0.8" "use_bias_correction=True" "safeguard_warmup=True" "betas=0.9,0.99" \
  --caption_extension=".txt" \
  --shuffle_caption \
  --keep_tokens=4 \
  --max_token_length=225 \
  --output_dir="/root/autodl-tmp/" \
  --output_name="Zhangshangyi_万物终归向海_何必携木石来" \
  --save_precision="fp16" \
  --save_every_n_epochs=1 \
  --train_batch_size=4 \
  --gradient_checkpointing \
  --gradient_accumulation_steps=1 \
  --max_train_epochs=20 \
  --seed=3407 \
  --xformers \
  --cache_latents \
  --cache_latents_to_disk \
  --persistent_data_loader_workers \
  --min_snr_gamma=5.0 \
  --noise_offset=0.1 \
  --no_half_vae \
  --logging_dir="/root/autodl-tmp/logging" \
  --log_with="tensorboard" \
  --sample_prompts="/root/autodl-tmp/sample_prompts.txt" \
  --sample_every_n_epochs=1 \
  --sample_sampler="euler"