The ultimate training toolkit for finetuning diffusion models - View it on GitHub