| #!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 |