lucidrains / mmdit
Implementation of a single layer of the MMDiT, proposed in Stable Diffusion 3, in Pytorch
☆255Updated 2 months ago
Related projects ⓘ
Alternatives and complementary repositories for mmdit
- Implementation of rectified flow and some of its followup research / improvements in Pytorch☆191Updated 2 weeks ago
- Scaling Diffusion Transformers with Mixture of Experts☆207Updated 2 months ago
- Implementation of Autoregressive Diffusion in Pytorch☆300Updated 2 weeks ago
- Rectified Diffusion: Straightness Is Not Your Need☆128Updated this week
- MoVQGAN - model for the image encoding and reconstruction☆197Updated last year
- Code for Fast Training of Diffusion Models with Masked Transformers☆373Updated 6 months ago
- This repo contains the code for 1D tokenizer and generator☆548Updated this week
- Train VAE like a boss☆246Updated 3 weeks ago
- Implementation of TiTok, proposed by Bytedance in "An Image is Worth 32 Tokens for Reconstruction and Generation"☆161Updated 5 months ago
- Scalable Diffusion Models with State Space Backbone☆146Updated 8 months ago
- [ICML 2024 Spotlight] FiT: Flexible Vision Transformer for Diffusion Model☆389Updated last week
- Official Pytorch Implementation of Representation Alignment for Generation: Training Diffusion Transformers Is Easier Than You Think☆664Updated this week
- ☆195Updated last month
- Implementation of a multimodal diffusion transformer in Pytorch☆97Updated 4 months ago
- Code for "Diffusion Model Alignment Using Direct Preference Optimization"☆269Updated 10 months ago
- Implementation of the proposed MaskBit from Bytedance AI☆62Updated last week
- Analyzing and Improving the Training Dynamics of Diffusion Models (EDM2)☆532Updated 5 months ago
- AlignProp uses direct reward backpropogation for the alignment of large-scale text-to-image diffusion models. Our method is 25x more samp…☆242Updated 2 weeks ago
- FMBoost: Boosting Latent Diffusion with Flow Matching (ECCV 2024 Oral)☆185Updated last month
- Official PyTorch Implementation of "SiT: Exploring Flow and Diffusion-based Generative Models with Scalable Interpolant Transformers"☆682Updated 8 months ago
- Official code for "DPM-Solver-v3: Improved Diffusion ODE Solver with Empirical Model Statistics" (NeurIPS 2023)☆99Updated 7 months ago
- Official Implementation for "Consistency Flow Matching: Defining Straight Flows with Velocity Consistency"☆151Updated 4 months ago
- Transformer-Mamba Diffusion Models☆89Updated 4 months ago
- Official PyTorch Implementation of "Scalable Autoregressive Image Generation with Mamba"☆110Updated 2 months ago
- [NeurIPS 2024] CV-VAE: A Compatible Video VAE for Latent Generative Video Models☆243Updated 2 weeks ago
- PeRFlow: Piecewise Rectified Flow as Universal Plug-and-Play Accelerator (NeurIPS 2024)☆445Updated 5 months ago
- HART: Efficient Visual Generation with Hybrid Autoregressive Transformer☆340Updated last month
- Consistency Models Made Easy☆226Updated last month
- Smooth Diffusion: Crafting Smooth Latent Spaces in Diffusion Models arXiv 2023 / CVPR 2024☆319Updated last month
- Masked Diffusion Transformer is the SOTA for image synthesis. (ICCV 2023)☆526Updated 6 months ago