lucidrains / mmdit
Implementation of a single layer of the MMDiT, proposed in Stable Diffusion 3, in Pytorch
☆351Updated 3 months ago
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☆285Updated 2 weeks ago
- Code for Fast Training of Diffusion Models with Masked Transformers☆401Updated 11 months ago
- Scaling Diffusion Transformers with Mixture of Experts☆317Updated 8 months ago
- Implementation of Autoregressive Diffusion in Pytorch☆376Updated 6 months ago
- [ICLR 2025] Rectified Diffusion: Straightness Is Not Your Need☆218Updated last month
- Official PyTorch Implementation of "SiT: Exploring Flow and Diffusion-based Generative Models with Scalable Interpolant Transformers"☆829Updated last year
- Code for "Diffusion Model Alignment Using Direct Preference Optimization"☆480Updated 3 months ago
- Train VAE like a boss☆277Updated 6 months ago
- [ICML 2024 Spotlight] FiT: Flexible Vision Transformer for Diffusion Model☆410Updated 6 months ago
- [ICLR'25 Oral] Representation Alignment for Generation: Training Diffusion Transformers Is Easier Than You Think☆1,016Updated last month
- Minimal implementation of scalable rectified flow transformers, based on SD3's approach☆519Updated 10 months ago
- PeRFlow: Piecewise Rectified Flow as Universal Plug-and-Play Accelerator (NeurIPS 2024)☆506Updated 11 months ago
- This repository includes the official implementation of our paper "Beyond Next-Token: Next-X Prediction for Autoregressive Visual Generat…☆203Updated last week
- MoVQGAN - model for the image encoding and reconstruction☆233Updated last year
- Consistency Models Made Easy☆277Updated 6 months ago
- This repo contains the code for 1D tokenizer and generator☆857Updated last month
- ☆484Updated 5 months ago
- Official implementation of the paper: REPA-E: Unlocking VAE for End-to-End Tuning of Latent Diffusion Transformers☆190Updated 3 weeks ago
- ☆184Updated 3 months ago
- DDT: Decoupled Diffusion Transformer☆233Updated 3 weeks ago
- Implementation of MagViT2 Tokenizer in Pytorch☆601Updated 3 months ago
- Official Implementation for Diffusion Models Without Classifier-free Guidance☆118Updated 2 months ago
- Official pytorch implementation of the paper: "An Edit Friendly DDPM Noise Space: Inversion and Manipulations". CVPR 2024.☆335Updated 9 months ago
- [CVPR2025] PyTorch-based reimplementation of CrossFlow, as proposed in 'Flowing from Words to Pixels: A Noise-Free Framework for Cross-Mo…☆167Updated last month
- AlignProp uses direct reward backpropogation for the alignment of large-scale text-to-image diffusion models. Our method is 25x more samp…☆282Updated 6 months ago
- Smooth Diffusion: Crafting Smooth Latent Spaces in Diffusion Models arXiv 2023 / CVPR 2024☆340Updated 7 months ago
- Official implementation of Inductive Moment Matching☆463Updated last month
- EDM2 and Autoguidance -- Official PyTorch implementation☆705Updated 5 months ago
- Scalable Diffusion Models with State Space Backbone☆152Updated last year
- FMBoost: Boosting Latent Diffusion with Flow Matching (ECCV 2024 Oral)☆226Updated 5 months ago