ry as the target images. These images are stored in class_data_dir.   Copied python retrieve.py --class_prompt cat --class_data_dir real_reg/samples_cat --num_class_images 200 To enable regularization, add the following parameters: --with_prior_preservation: whether to use prior preservation loss --prior_loss_weight: controls the influence of the prior preservation loss on the model --real_prior: whether to use a small set of real images to pre