<Tip>  
A lot of the code in the Custom Diffusion training script is similar to the [DreamBooth](dreambooth#training-script) script. This guide instead focuses on the code that is relevant to Custom Diffusion.  
</Tip>  
The Custom Diffusion training script has two dataset classes: