="cat" \
  --real_prior=True \  Training script A lot of the code in the Custom Diffusion training script is similar to the DreamBooth script. This guide instead focuses on the code that is relevant to Custom Diffusion. The Custom Diffusion training script has two dataset classes: CustomDiffusionDataset: preprocesses the images, class images, and prompts for training PromptDataset: prepares the prompts for generating class images Next, the modi