mages to a local path you specify.
The authors recommend generating num_epochs * num_samples images for prior preservation. In most cases, 200-300 images work well.


Pytorch

Hide Pytorch content



	Copied
export MODEL_NAME="CompVis/stable-diffusion-v1-4"
export INSTANCE_DIR="./dog"
export CLASS_DIR="path_to_class_images"
export OUTPUT_DIR="path_to_saved_model"

accelerate launch train_dreambooth.py \
  --pretrained_model_name_or_path=$MODEL_