inspired by https://github.com/mikonvergence/DiffusionFastForward- Check it out for an in-depth video tutorial + notebooks.
Both the train + train_conditional as well as the notebook hold examples that can be run on your own data, just supplement your DataLoader to get going.
https://colab.research.google.com/drive/1-NW6_HiqA3Ox8nvTrBnOK9-rHQGo6jzT?usp=sharing
https://drive.google.com/file/d/1tkSy9DQQE0g3j2KnUG1VTGrXe2Zxi7y1/view?usp=drive_link
https://docs.google.com/presentation/d/1-hJdELN5yZziS0J21oSlo-PecgWVW5leAxTNBo4sfQQ/edit?usp=sharing