The ultimate training toolkit for finetuning diffusion models