Implementation of a multimodal diffusion transformer in Pytorch - View it on GitHub