File size: 517 Bytes
c30c4ee |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
#!usr/bin/bash
accelerate launch --num_cpu_threads_per_process 1 train_db.py \
--pretrained_model_name_or_path="runwayml/stable-diffusion-v1-5" \
--dataset_config=config.toml \
--output_dir=db-kohya-tuandao-folder \
--output_name=db-kohya-tuandao \
--save_model_as=safetensors \
--prior_loss_weight=1.0 \
--max_train_steps=400 \
--learning_rate=1e-6 \
--optimizer_type="AdamW8bit" \
--xformers \
--mixed_precision="fp16" \
--cache_latents \
--gradient_checkpointing |