export LD_LIBRARY_PATH=/usr/lib/wsl/lib:$LD_LIBRARY_PATH
export MODEL_NAME="runwayml/stable-diffusion-v1-5"
export OUTPUT_DIR="/mnt/c/temp/ft-trek"
export DATA_DIR="/mnt/c/temp/trek-ds/"

accelerate launch --num_cpu_threads_per_process 6 train_db_fixed.py \
    --pretrained_model_name_or_path=$MODEL_NAME \
    --train_data_dir=$DATA_DIR \
    --gradient_checkpointing \
    --shuffle_caption \
    --save_state \
    --fine_tuning \
    --use_safetensors \
    --output_dir=$OUTPUT_DIR \
    --resolution="512,512" \
    --train_batch_size=2 \
    --learning_rate=1e-7 \
    --use_8bit_adam \
    --xformers \
    --mixed_precision="fp16" \
    --cache_latents \
    --save_every_n_epochs=2 \
    --caption_extension=".txt" \
    --enable_bucket \
    --dataset_repeats=10 \
    --seed=5318008 \
    --save_precision="fp16" \