dreambooth_face_01

# Sample .sh script

export MODEL_NAME="runwayml/stable-diffusion-v1-5"
export INSTANCE_DIR="training_kasuka"
export CLASS_DIR="classes_kasuka"
# export OUTPUT_DIR="model_kasuka"
export OUTPUT_DIR="model_kasuka_02"

let NUM_INSTANCE_IMAGES=11
let NUM_CLASS_IMAGES=50
let MAX_NUM_STEPS=NUM_INSTANCE_IMAGES*80
LR_SCHEDULE="constant"
let LR_WARMUP_STEPS=MAX_NUM_STEPS/10
let CHECKPOINT_SAVE_STEPS=MAX_NUM_STEPS/4

echo $NUM_INSTANCE_IMAGES
echo $NUM_CLASS_IMAGES
echo $MAX_NUM_STEPS
echo $LR_WARMUP_STEPS
echo $LR_SCHEDULE

accelerate launch train_dreambooth.py \
  --pretrained_model_name_or_path=$MODEL_NAME \
  --instance_data_dir=$INSTANCE_DIR \
  --class_data_dir=$CLASS_DIR \
  --instance_prompt="photo of k4suk4nr4 person" \
  --class_prompt="photo of a person" \
  --output_dir=$OUTPUT_DIR \
  --revision="fp16" \
  --with_prior_preservation --prior_loss_weight=1.0 \
  --seed=1337 \
  --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 \
  --num_class_images=$NUM_CLASS_IMAGES \
  --max_train_steps=$MAX_NUM_STEPS \
  --checkpointing_steps=$CHECKPOINT_SAVE_STEPS