/U-ViT

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

Primary LanguageJupyter NotebookMIT LicenseMIT

This repository is not active