lumalabs / imm
Official implementation of Inductive Moment Matching
☆458Updated last month
Alternatives and similar repositories for imm:
Users that are interested in imm are comparing it to the libraries listed below
- Minimal implementation of scalable rectified flow transformers, based on SD3's approach☆517Updated 10 months ago
- Train VAE like a boss☆276Updated 6 months ago
- PeRFlow: Piecewise Rectified Flow as Universal Plug-and-Play Accelerator (NeurIPS 2024)☆506Updated 11 months ago
- DDT: Decoupled Diffusion Transformer☆233Updated 3 weeks ago
- (CVPR 2025) From Slow Bidirectional to Fast Autoregressive Video Diffusion Models☆257Updated 3 weeks ago
- Official PyTorch Implementation of "History-Guided Video Diffusion"☆296Updated 2 months ago
- FMBoost: Boosting Latent Diffusion with Flow Matching (ECCV 2024 Oral)☆226Updated 5 months ago
- Implementation of a single layer of the MMDiT, proposed in Stable Diffusion 3, in Pytorch☆351Updated 3 months ago
- FlexTok: Resampling Images into 1D Token Sequences of Flexible Length☆126Updated last month
- [CVPR2025] PyTorch-based reimplementation of CrossFlow, as proposed in 'Flowing from Words to Pixels: A Noise-Free Framework for Cross-Mo…☆167Updated last month
- Implementation of rectified flow and some of its followup research / improvements in Pytorch☆285Updated 2 weeks ago
- Block Diffusion: Interpolating Between Autoregressive and Diffusion Language Models☆569Updated 3 weeks ago
- ☆484Updated 5 months ago
- HART: Efficient Visual Generation with Hybrid Autoregressive Transformer☆575Updated 6 months ago
- ☆193Updated 2 months ago
- EDM2 and Autoguidance -- Official PyTorch implementation☆705Updated 5 months ago
- code for "Diffusion Forcing: Next-token Prediction Meets Full-Sequence Diffusion"☆843Updated last month
- [ICLR 2025] Rectified Diffusion: Straightness Is Not Your Need☆216Updated last month
- (NeurIPS 2024 Oral 🔥) Improved Distribution Matching Distillation for Fast Image Synthesis☆769Updated 2 months ago
- [ICLR 2025] Autoregressive Video Generation without Vector Quantization☆492Updated 2 weeks ago
- ☆178Updated 4 months ago
- T-GATE: Temporally Gating Attention to Accelerate Diffusion Model for Free!☆395Updated 2 months ago
- The aim of this repository is to test and implement Flow-Matching-based models☆91Updated 3 months ago
- Pixel-Space Generative Models☆197Updated this week
- Official implementation of the paper: REPA-E: Unlocking VAE for End-to-End Tuning of Latent Diffusion Transformers☆190Updated 3 weeks ago
- Inference-time scaling of diffusion-based image and video generation models.☆142Updated 2 months ago
- Official code for "RB-Modulation: Training-Free Personalization of Diffusion Models using Stochastic Optimal Control"☆379Updated last month
- [WACV'25 Oral] Fine-Tuning Image-Conditional Diffusion Models is Easier than You Think☆422Updated 4 months ago
- [NeurIPS 2024] Boosting the performance of consistency models with PCM!☆469Updated 4 months ago
- [ICLR'25 Oral] Representation Alignment for Generation: Training Diffusion Transformers Is Easier Than You Think☆1,016Updated last month