#!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=1600 
    --learning_rate=1e-6 
    --optimizer_type="AdamW8bit" 
    --xformers 
    --mixed_precision="fp16" 
    --cache_latents 
    --gradient_checkpointing