baofff / U-ViT

A PyTorch implementation of the paper "All are Worth Words: A ViT Backbone for Diffusion Models".
MIT License
888 stars 58 forks source link

[New Feature] diffusers support #6

Open Adamdad opened 1 year ago

Adamdad commented 1 year ago

Do you have the plan to support the U-ViT for diffusers?