export LD_LIBRARY_PATH=/usr/lib/wsl/lib:$LD_LIBRARY_PATH

export MODEL_NAME="/home/systemy/NAI"
export OUTPUT_DIR="/mnt/d/jigsaw"

accelerate launch train_dreambooth.py \
  --pretrained_model_name_or_path="$MODEL_NAME" \
  --pretrained_vae_name_or_path="stabilityai/sd-vae-ft-mse" \
  --output_dir="/mnt/d/acuteoutput" \
  --seed=3434554 \
  --resolution=512 \
  --train_batch_size=1 \
  --train_text_encoder \
  --mixed_precision="fp16" \
  --use_8bit_adam \
  --gradient_accumulation_steps=1 \
  --learning_rate=1e-6 \
  --lr_scheduler="constant" \
  --lr_warmup_steps=0 \
  --sample_batch_size=6 \
  --max_train_steps=100000 \
  --save_interval=1500 \
  --save_sample_prompt="image of SystemyTrigger girl" \
  --concepts_list="concepts_list.json" \
  --pad_tokens