lucidrains / mmditLinks
Implementation of a single layer of the MMDiT, proposed in Stable Diffusion 3, in Pytorch
☆369Updated 5 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☆309Updated last week
- Code for Fast Training of Diffusion Models with Masked Transformers☆403Updated last year
- Official PyTorch Implementation of "SiT: Exploring Flow and Diffusion-based Generative Models with Scalable Interpolant Transformers"☆873Updated last year
- Scaling Diffusion Transformers with Mixture of Experts☆339Updated 9 months ago
- MoVQGAN - model for the image encoding and reconstruction☆240Updated last year
- Implementation of Autoregressive Diffusion in Pytorch☆390Updated 7 months ago
- Implementation of MagViT2 Tokenizer in Pytorch☆608Updated 5 months ago
- Code for "Diffusion Model Alignment Using Direct Preference Optimization"☆504Updated 4 months ago
- This repo contains the code for 1D tokenizer and generator☆918Updated 3 months ago
- SEED-Voken: A Series of Powerful Visual Tokenizers☆897Updated 4 months ago
- DDT: Decoupled Diffusion Transformer☆257Updated last week
- Official implementation of the paper: REPA-E: Unlocking VAE for End-to-End Tuning of Latent Diffusion Transformers☆263Updated 2 months ago
- [ICLR'25 Oral] Representation Alignment for Generation: Training Diffusion Transformers Is Easier Than You Think☆1,126Updated 3 months ago
- EDM2 and Autoguidance -- Official PyTorch implementation☆729Updated 6 months ago
- [ICLR 2025] Rectified Diffusion: Straightness Is Not Your Need☆229Updated 3 months ago
- [CVPR2025] PyTorch-based reimplementation of CrossFlow, as proposed in 'Flowing from Words to Pixels: A Noise-Free Framework for Cross-Mo…☆262Updated 2 weeks ago
- PeRFlow: Piecewise Rectified Flow as Universal Plug-and-Play Accelerator (NeurIPS 2024)☆519Updated last year
- [ICML 2024 Spotlight] FiT: Flexible Vision Transformer for Diffusion Model☆416Updated 7 months ago
- This repository includes the official implementation of our paper "Beyond Next-Token: Next-X Prediction for Autoregressive Visual Generat…☆215Updated last month
- An official implementation of Flow-GRPO: Training Flow Matching Models via Online RL☆783Updated last week
- ☆291Updated 8 months ago
- Train VAE like a boss☆281Updated 8 months ago
- Consistency Models Made Easy☆289Updated 8 months ago
- Minimal implementation of scalable rectified flow transformers, based on SD3's approach☆589Updated 11 months ago
- [ECCV 2024] Official Repository for DiffiT: Diffusion Vision Transformers for Image Generation☆495Updated 7 months ago
- ☆549Updated 6 months ago
- ☆204Updated 3 weeks ago
- ☆200Updated 6 months ago
- AlignProp uses direct reward backpropogation for the alignment of large-scale text-to-image diffusion models. Our method is 25x more samp…☆289Updated 7 months ago
- Official implementation of Inductive Moment Matching☆488Updated 3 months ago