A mini-library for training consistency models.
☆251Dec 26, 2023Updated 2 years ago
Alternatives and similar repositories for consistency_models
Users that are interested in consistency_models are comparing it to the libraries listed below
Sorting:
- Consistency Models Made Easy☆325Oct 13, 2024Updated last year
- Unofficial Implementation of Consistency Models in pytorch☆260Mar 7, 2023Updated 2 years ago
- A Toolkit for OpenAI's Consistency Models.☆207Apr 14, 2023Updated 2 years ago
- ☆318Nov 17, 2025Updated 3 months ago
- Official repo for consistency models.☆6,477Mar 22, 2024Updated last year
- Implementation of Consistency Models (Song et al 2023) for few-step image generation in Jax.☆19Jun 11, 2023Updated 2 years ago
- Elucidating the Design Space of Diffusion-Based Generative Models (EDM)☆1,915Mar 16, 2024Updated last year
- Implementation of "Analyzing and Improving the Training Dynamics of Diffusion Models"☆97Feb 12, 2024Updated 2 years ago
- Minimum implementation of EDM (Elucidating the Design Space of Diffusion-Based Generative Models) on cifar10 and mnist☆64Dec 16, 2023Updated 2 years ago
- EDM2 and Autoguidance -- Official PyTorch implementation☆822Dec 9, 2024Updated last year
- Awesome List of Consistency Models☆131Dec 9, 2025Updated 2 months ago
- Official PyTorch implementation for the paper Minimizing Trajectory Curvature of ODE-based Generative Models, ICML 2023☆91Feb 14, 2025Updated last year
- Unofficial minimal implementation of consistency models (CM) proposed by Song et al. 2023 on a 1D toy task in pytorch☆21May 2, 2023Updated 2 years ago
- A JAX implementation of the continuous time formulation of Consistency Models☆85Apr 7, 2023Updated 2 years ago
- Official Implementation of Rectified Flow (ICLR2023 Spotlight)☆1,552Jul 20, 2024Updated last year
- Official Implementation for "Consistency Flow Matching: Defining Straight Flows with Velocity Consistency"☆259Jan 17, 2025Updated last year
- Consistency models trained on CIFAR-10, in JAX.☆151Aug 22, 2023Updated 2 years ago
- [ICML 2023] Reflected Diffusion Models (https://arxiv.org/abs/2304.04740)☆159Oct 19, 2023Updated 2 years ago
- TorchCFM: a Conditional Flow Matching library☆2,319Nov 11, 2025Updated 3 months ago
- Official PyTorch Implementation of "SiT: Exploring Flow and Diffusion-based Generative Models with Scalable Interpolant Transformers"☆1,096Dec 22, 2025Updated 2 months ago
- Official implementation for: Consistency Models as a Rich and Efficient Policy Class for Reinforcement Learning ICLR'24☆26Aug 28, 2024Updated last year
- PyTorch implementation for Score-Based Generative Modeling through Stochastic Differential Equations (ICLR 2021, Oral)☆2,082Jul 14, 2024Updated last year
- PyTorch Implementation of Diffusion Schrodinger Bridge Matching☆202Feb 19, 2025Updated last year
- [ICML 2024 Best Paper] Discrete Diffusion Modeling by Estimating the Ratios of the Data Distribution (https://arxiv.org/abs/2310.16834)☆707Feb 29, 2024Updated 2 years ago
- The Official PyTorch Implementation of "LSGM: Score-based Generative Modeling in Latent Space" (NeurIPS 2021)☆370Dec 2, 2021Updated 4 years ago
- Official code for Score-Based Generative Modeling through Stochastic Differential Equations (ICLR 2021, Oral)☆1,802Nov 29, 2022Updated 3 years ago
- The codebase of our paper "Improving the Training of Rectified Flows", NeurIPS 2024☆130Oct 18, 2024Updated last year
- [NeurIPS 2023] Official Implementation: "Consistent Diffusion Models"☆58Apr 17, 2023Updated 2 years ago
- Official pytorch repository for "Diffusion Posterior Sampling for General Noisy Inverse Problems"☆617Mar 2, 2023Updated 3 years ago
- The official PyTorch implementation of Fast Diffusion Model☆96Jun 27, 2023Updated 2 years ago
- Masked Diffusion Transformer is the SOTA for image synthesis. (ICCV 2023)☆593Apr 23, 2024Updated last year
- ☆739Dec 5, 2024Updated last year
- [ICCV 2023] Efficient Diffusion Training via Min-SNR Weighting Strategy☆268Dec 10, 2024Updated last year
- Minimal implementation of scalable rectified flow transformers, based on SD3's approach☆634Jul 1, 2024Updated last year
- Score-Based Generative Modeling with Critically-Damped Langevin Diffusion☆201Oct 23, 2023Updated 2 years ago
- Karras et al. (2022) diffusion models for PyTorch☆2,567Feb 12, 2026Updated 2 weeks ago
- Consistency Distilled Diff VAE☆2,209Nov 7, 2023Updated 2 years ago
- Code for Fast Training of Diffusion Models with Masked Transformers☆421May 15, 2024Updated last year
- ☆7,305Jul 2, 2024Updated last year