lucidrains / mmditLinks
Implementation of a single layer of the MMDiT, proposed in Stable Diffusion 3, in Pytorch
☆361Updated 4 months ago
Alternatives and similar repositories for mmdit
Users that are interested in mmdit are comparing it to the libraries listed below
Sorting:
- Implementation of rectified flow and some of its followup research / improvements in Pytorch☆292Updated last month
- Code for Fast Training of Diffusion Models with Masked Transformers☆403Updated last year
- Scaling Diffusion Transformers with Mixture of Experts☆330Updated 8 months ago
- Official PyTorch Implementation of "SiT: Exploring Flow and Diffusion-based Generative Models with Scalable Interpolant Transformers"☆847Updated last year
- PeRFlow: Piecewise Rectified Flow as Universal Plug-and-Play Accelerator (NeurIPS 2024)☆512Updated last year
- MoVQGAN - model for the image encoding and reconstruction☆240Updated last year
- Implementation of Autoregressive Diffusion in Pytorch☆382Updated 7 months ago
- [ICLR'25 Oral] Representation Alignment for Generation: Training Diffusion Transformers Is Easier Than You Think☆1,062Updated 2 months ago
- This repo contains the code for 1D tokenizer and generator☆887Updated 2 months ago
- ☆530Updated 5 months ago
- [ICML 2024 Spotlight] FiT: Flexible Vision Transformer for Diffusion Model☆409Updated 6 months ago
- Code for "Diffusion Model Alignment Using Direct Preference Optimization"☆492Updated 4 months ago
- EDM2 and Autoguidance -- Official PyTorch implementation☆714Updated 5 months ago
- Consistency Models Made Easy☆284Updated 7 months ago
- FMBoost: Boosting Latent Diffusion with Flow Matching (ECCV 2024 Oral)☆231Updated 6 months ago
- [ECCV 2024] Official Repository for DiffiT: Diffusion Vision Transformers for Image Generation☆492Updated 7 months ago
- Implementation of MagViT2 Tokenizer in Pytorch☆605Updated 4 months ago
- [ICLR 2025] Rectified Diffusion: Straightness Is Not Your Need☆224Updated 2 months ago
- An official implementation of Flow-GRPO: Training Flow Matching Models via Online RL☆681Updated 2 weeks ago
- This repository includes the official implementation of our paper "Beyond Next-Token: Next-X Prediction for Autoregressive Visual Generat…☆207Updated last month
- Train VAE like a boss☆279Updated 7 months ago
- Official implementation of the paper: REPA-E: Unlocking VAE for End-to-End Tuning of Latent Diffusion Transformers☆233Updated last month
- Pytorch Implementation (unofficial) of the paper "Mean Flows for One-step Generative Modeling" by Geng et al.☆245Updated this week
- DDT: Decoupled Diffusion Transformer☆252Updated 2 weeks ago
- Minimal implementation of scalable rectified flow transformers, based on SD3's approach☆562Updated 11 months ago
- Scalable Diffusion Models with State Space Backbone☆154Updated last year
- SEED-Voken: A Series of Powerful Visual Tokenizers☆885Updated 3 months ago
- [ICLR 2025] Autoregressive Video Generation without Vector Quantization☆510Updated last week
- ☆289Updated 7 months ago
- ☆193Updated this week