NVlabs / edm2
Analyzing and Improving the Training Dynamics of Diffusion Models (EDM2)
☆532Updated 5 months ago
Related projects ⓘ
Alternatives and complementary repositories for edm2
- ☆239Updated last month
- Code for Fast Training of Diffusion Models with Masked Transformers☆375Updated 6 months ago
- Consistency Models Made Easy☆226Updated last month
- A summary of related works about flow matching, stochastic interpolants☆337Updated 3 months ago
- A mini-library for training consistency models.☆215Updated 10 months ago
- Official Pytorch Implementation of Representation Alignment for Generation: Training Diffusion Transformers Is Easier Than You Think☆669Updated last week
- [ECCV 2024] Official Repository for DiffiT: Diffusion Vision Transformers for Image Generation☆459Updated 3 weeks ago
- Official PyTorch Implementation of "SiT: Exploring Flow and Diffusion-based Generative Models with Scalable Interpolant Transformers"☆689Updated 8 months ago
- Implementation of rectified flow and some of its followup research / improvements in Pytorch☆195Updated this week
- [ICML 2024 Best Paper] Discrete Diffusion Modeling by Estimating the Ratios of the Data Distribution (https://arxiv.org/abs/2310.16834)☆408Updated 8 months ago
- [ICCV 2023] Efficient Diffusion Training via Min-SNR Weighting Strategy☆229Updated 7 months ago
- Official Implementation of Rectified Flow (ICLR2023 Spotlight)☆960Updated 4 months ago
- ☆456Updated last year
- Official PyTorch implementation of the paper: Flow Matching in Latent Space☆209Updated 3 weeks ago
- Code for ICML 2023 paper, "PFGM++: Unlocking the Potential of Physics-Inspired Generative Models"☆360Updated last year
- Implementation of a single layer of the MMDiT, proposed in Stable Diffusion 3, in Pytorch☆255Updated 2 months ago
- Implementation of Autoregressive Diffusion in Pytorch☆302Updated 2 weeks ago
- ☆211Updated last year
- Elucidating the Design Space of Diffusion-Based Generative Models (EDM)☆1,400Updated 8 months ago
- Train VAE like a boss☆247Updated last month
- Unofficial Implementation of Consistency Models in pytorch☆251Updated last year
- Official Jax Implementation of MaskGIT☆450Updated 2 years ago
- Masked Diffusion Transformer is the SOTA for image synthesis. (ICCV 2023)☆528Updated 6 months ago
- Text to Image Latent Diffusion using a Transformer core☆145Updated 2 months ago
- Minimal implementation of scalable rectified flow transformers, based on SD3's approach☆444Updated 4 months ago
- code for "Diffusion Forcing: Next-token Prediction Meets Full-Sequence Diffusion"☆616Updated last week
- ☆259Updated last year
- PyTorch implementation of MAR+DiffLoss https://arxiv.org/abs/2406.11838☆1,020Updated last month
- DDPO for finetuning diffusion models, implemented in PyTorch with LoRA support☆434Updated 8 months ago
- This repo contains the code for 1D tokenizer and generator☆554Updated this week