byeongjun-park / Switch-DiT

[ECCV 2024] Official pytorch implementation of "Switch Diffusion Transformer: Synergizing Denoising Tasks with Sparse Mixture-of-Experts"
30Updated 2 months ago

Related projects: