The implementation seems to be pretty clean and modular here where transformers (and diffusers) isn’t, unless you take their modules standalone. This repo has

The implementation seems to be pretty clean and modular here where transformers (and diffusers) isn’t, unless you take their modules standalone. This repo has a lot of handy utilities but also a bunch of clean implementations of common models, metrics, etc. In other words, this is more for writing new models rather than inference.