elcome a Feature Request or directly a Pull Request from you!
Training examples show how to pretrain or fine-tune diffusion models for a variety of tasks. Currently we support:
Unconditional Training
Text-to-Image Training
Text Inversion
Dreambooth
LoRA Support
ControlNet
InstructPix2Pix
Custom Diffusion
If possible, please install xFormers for memory efficient attention. This could help make your training faster and less memory intensive.
Task