export MODEL_NAME="stable-diffusion/stable-diffusion-1-5"
export INSTANCE_DIR="data/training/"
export OUTPUT_DIR="output/lora/"
export CLASS_DIR="data/lora_class"

accelerate launch eval/train_dreambooth_lora_15.py \
  --pretrained_model_name_or_path=$MODEL_NAME  \
  --instance_data_dir=$INSTANCE_DIR \
  --output_dir=$OUTPUT_DIR \
  --class_data_dir=$CLASS_DIR\
  --instance_prompt "an animated painting of a sks person, high quality, masterpiece" \
  --class_prompt "an animated painting of a person, high quality, masterpiece" \
  --resolution=512 \
  --train_batch_size=1 \
  --gradient_accumulation_steps=1 \
  --learning_rate=1e-4 \
  --scale_lr \
  --max_train_steps=2000 \
  --mixed_precision=bf16 \