GitHub - baofff/U-ViT: A PyTorch implementation of the paper "All are Worth Words: A ViT Backbone for Diffusion Models".

A PyTorch implementation of the paper "All are Worth Words: A ViT Backbone for Diffusion Models". - baofff/U-ViT