lucidrains / mmdit
Implementation of a single layer of the MMDiT, proposed in Stable Diffusion 3, in Pytorch
☆307Updated last month
Alternatives and similar repositories for mmdit:
Users that are interested in mmdit are comparing it to the libraries listed below
- Implementation of rectified flow and some of its followup research / improvements in Pytorch☆247Updated last week
- Scaling Diffusion Transformers with Mixture of Experts☆260Updated 5 months ago
- Code for Fast Training of Diffusion Models with Masked Transformers☆388Updated 9 months ago
- Official PyTorch Implementation of "SiT: Exploring Flow and Diffusion-based Generative Models with Scalable Interpolant Transformers"☆764Updated 11 months ago
- Implementation of Autoregressive Diffusion in Pytorch☆356Updated 3 months ago
- [ICLR 2025] Rectified Diffusion: Straightness Is Not Your Need☆183Updated 2 months ago
- ☆376Updated 2 months ago
- MoVQGAN - model for the image encoding and reconstruction☆218Updated last year
- Train VAE like a boss☆265Updated 4 months ago
- Code for "Diffusion Model Alignment Using Direct Preference Optimization"☆352Updated 2 weeks ago
- Minimal implementation of scalable rectified flow transformers, based on SD3's approach☆478Updated 7 months ago
- Implementation of MagViT2 Tokenizer in Pytorch☆590Updated last month
- [ICML 2024 Spotlight] FiT: Flexible Vision Transformer for Diffusion Model☆401Updated 3 months ago
- Official Pytorch Implementation of Representation Alignment for Generation: Training Diffusion Transformers Is Easier Than You Think (ICL…☆832Updated 3 weeks ago
- This repo contains the code for 1D tokenizer and generator☆691Updated last week
- [arXiv'25] Reconstruction vs. Generation: Taming Optimization Dilemma in Latent Diffusion Models☆251Updated last month
- HART: Efficient Visual Generation with Hybrid Autoregressive Transformer☆418Updated 4 months ago
- PeRFlow: Piecewise Rectified Flow as Universal Plug-and-Play Accelerator (NeurIPS 2024)☆484Updated 8 months ago
- SEED-Voken: A Series of Powerful Visual Tokenizers☆830Updated this week
- Consistency Models Made Easy☆262Updated 4 months ago
- Scalable Diffusion Models with State Space Backbone☆151Updated 11 months ago
- [NeurIPS 2024] The official code of "U-DiTs: Downsample Tokens in U-Shaped Diffusion Transformers"☆182Updated 4 months ago
- ☆136Updated last week
- Smooth Diffusion: Crafting Smooth Latent Spaces in Diffusion Models arXiv 2023 / CVPR 2024☆330Updated 4 months ago
- [ECCV 2024] Official Repository for DiffiT: Diffusion Vision Transformers for Image Generation☆487Updated 3 months ago
- EDM2 and Autoguidance -- Official PyTorch implementation☆632Updated 2 months ago
- Masked Diffusion Transformer is the SOTA for image synthesis. (ICCV 2023)☆547Updated 9 months ago
- Official Implementation for "Consistency Flow Matching: Defining Straight Flows with Velocity Consistency"☆192Updated last month
- Implementation of a multimodal diffusion transformer in Pytorch☆100Updated 7 months ago
- ☆71Updated 4 months ago