baofff / U-ViT

A PyTorch implementation of the paper "All are Worth Words: A ViT Backbone for Diffusion Models".
924Updated last year

Related projects

Alternatives and complementary repositories for U-ViT