use for regularization:   Copied pip install clip-retrieval
python retrieve.py --class_prompt {} --class_data_dir {} --num_class_images 200 Then you can launch the script:   Copied export MODEL_NAME="CompVis/stable-diffusion-v1-4"
export OUTPUT_DIR="path-to-save-model"

accelerate launch train_custom_diffusion.py \
  --pretrained_model_name_or_path=$MODEL_NAME  \
  --output_dir=$OUTPUT_DIR \
  --concepts_list=./concept_list.json \
  --with_prio