lucidrains / mmdit
Implementation of a single layer of the MMDiT, proposed in Stable Diffusion 3, in Pytorch
☆338Updated 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☆279Updated 2 months ago
- Scaling Diffusion Transformers with Mixture of Experts☆311Updated 7 months ago
- Official PyTorch Implementation of "SiT: Exploring Flow and Diffusion-based Generative Models with Scalable Interpolant Transformers"☆807Updated last year
- Code for Fast Training of Diffusion Models with Masked Transformers☆398Updated 11 months ago
- [ICLR 2025] Rectified Diffusion: Straightness Is Not Your Need☆214Updated last month
- Code for "Diffusion Model Alignment Using Direct Preference Optimization"☆450Updated 2 months ago
- [ICLR'25 Oral] Representation Alignment for Generation: Training Diffusion Transformers Is Easier Than You Think☆930Updated last month
- PeRFlow: Piecewise Rectified Flow as Universal Plug-and-Play Accelerator (NeurIPS 2024)☆497Updated 10 months ago
- This repo contains the code for 1D tokenizer and generator☆832Updated last month
- ☆447Updated 4 months ago
- Implementation of Autoregressive Diffusion in Pytorch☆370Updated 5 months ago
- [ICML 2024 Spotlight] FiT: Flexible Vision Transformer for Diffusion Model☆411Updated 5 months ago
- MoVQGAN - model for the image encoding and reconstruction☆231Updated last year
- Train VAE like a boss☆274Updated 5 months ago
- Implementation of MagViT2 Tokenizer in Pytorch☆599Updated 3 months ago
- SEED-Voken: A Series of Powerful Visual Tokenizers☆865Updated 2 months ago
- Masked Diffusion Transformer is the SOTA for image synthesis. (ICCV 2023)☆559Updated 11 months ago
- EDM2 and Autoguidance -- Official PyTorch implementation☆692Updated 4 months ago
- AlignProp uses direct reward backpropogation for the alignment of large-scale text-to-image diffusion models. Our method is 25x more samp…☆282Updated 5 months ago
- Consistency Models Made Easy☆278Updated 6 months ago
- ☆177Updated 2 months ago
- Official Implementation for "Consistency Flow Matching: Defining Straight Flows with Velocity Consistency"☆207Updated 3 months ago
- Scalable Diffusion Models with State Space Backbone☆152Updated last year
- [ECCV 2024] Official Repository for DiffiT: Diffusion Vision Transformers for Image Generation☆491Updated 5 months ago
- This repository includes the official implementation of our paper "Beyond Next-Token: Next-X Prediction for Autoregressive Visual Generat…☆183Updated last month
- [ICLR 2025] Autoregressive Video Generation without Vector Quantization☆468Updated 2 weeks ago
- [ICLR2025] Halton Scheduler for Masked Generative Image Transformer☆217Updated last week
- This is a repo to track the latest autoregressive visual generation papers.☆257Updated this week
- Official pytorch implementation of the paper: "An Edit Friendly DDPM Noise Space: Inversion and Manipulations". CVPR 2024.☆332Updated 9 months ago
- FMBoost: Boosting Latent Diffusion with Flow Matching (ECCV 2024 Oral)☆227Updated 4 months ago