byeongjun-park/Switch-DiT

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

语言：Python · GitHub 星标：49 · Fork：9

话题：diffusion-models · mixture-of-experts

在 GitHub 上查看原仓库