script/run.sh

#!/bin/bash


 accelerate launch train.py \
 --pretrained_model_name_or_path="SG161222/Realistic_Vision_V5.1_noVAE" \
 --output_dir="./controlnet-training-runs" \
 --train_data_dir=/path/to/dataset \
 --cfg_prob=0.1 \
 --resolution=512 \
 --learning_rate=1e-5 \
  --num_validation_images=3 \
 --validation_image  "./face_normal.png"  "./face_seg.png"    \
 --validation_prompt  "a Teen boy, pensive look, dark hair. Preppy sweater, collared shirt, moody room, 80s memorabilia"   \
 --train_batch_size=4 \
 --num_train_epochs=40 \
 --validation_steps=500  \
 --checkpointing_steps=2000