Bringing together invertible UNets with invertible attention modules for memory-efficient diffusion models

Karan Jain, Mohammad Nayeem Teli

Diffusion models have recently gained state of the art performance on many image generation tasks. However, most models require significant computational resources to achieve this.…