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. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Consistency Models Made Easy☆328Oct 13, 2024Updated last year
- Unofficial Implementation of Consistency Models in pytorch☆260Mar 7, 2023Updated 3 years ago
- A Toolkit for OpenAI's Consistency Models.☆207Apr 14, 2023Updated 2 years ago
- Implementation of Consistency Models (Song et al 2023) for few-step image generation in Jax.☆19Jun 11, 2023Updated 2 years ago
- ☆319Mar 3, 2026Updated 2 weeks ago
- Minimum implementation of EDM (Elucidating the Design Space of Diffusion-Based Generative Models) on cifar10 and mnist☆64Dec 16, 2023Updated 2 years ago
- Official repo for consistency models.☆6,474Mar 22, 2024Updated 2 years ago
- Elucidating the Design Space of Diffusion-Based Generative Models (EDM)☆1,921Mar 16, 2024Updated 2 years ago
- Implementation of "Analyzing and Improving the Training Dynamics of Diffusion Models"☆97Feb 12, 2024Updated 2 years ago
- Awesome List of Consistency Models☆131Dec 9, 2025Updated 3 months ago
- A JAX implementation of the continuous time formulation of Consistency Models☆85Apr 7, 2023Updated 2 years ago
- EDM2 and Autoguidance -- Official PyTorch implementation☆824Dec 9, 2024Updated last year
- Consistency models trained on CIFAR-10, in JAX.☆153Aug 22, 2023Updated 2 years ago
- Official Implementation for "Consistency Flow Matching: Defining Straight Flows with Velocity Consistency"☆262Jan 17, 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
- Official implementation for: Consistency Models as a Rich and Efficient Policy Class for Reinforcement Learning ICLR'24☆26Aug 28, 2024Updated last year
- Official Implementation of Rectified Flow (ICLR2023 Spotlight)☆1,563Jul 20, 2024Updated last year
- [ICML 2023] Reflected Diffusion Models (https://arxiv.org/abs/2304.04740)☆159Oct 19, 2023Updated 2 years ago
- Official PyTorch implementation for the paper Minimizing Trajectory Curvature of ODE-based Generative Models, ICML 2023☆91Feb 14, 2025Updated last year
- PyTorch implementation for Score-Based Generative Modeling through Stochastic Differential Equations (ICLR 2021, Oral)☆2,089Jul 14, 2024Updated last year
- TorchCFM: a Conditional Flow Matching library☆2,357Nov 11, 2025Updated 4 months ago
- Official PyTorch Implementation of "SiT: Exploring Flow and Diffusion-based Generative Models with Scalable Interpolant Transformers"☆1,120Dec 22, 2025Updated 3 months ago
- Official pytorch repository for "Diffusion Posterior Sampling for General Noisy Inverse Problems"☆621Mar 2, 2023Updated 3 years ago
- The codebase of our paper "Improving the Training of Rectified Flows", NeurIPS 2024☆130Oct 18, 2024Updated last year
- Official code for Score-Based Generative Modeling through Stochastic Differential Equations (ICLR 2021, Oral)☆1,811Nov 29, 2022Updated 3 years ago
- Official repo for "Solving Inverse Problems in Medical Imaging with Score-Based Generative Models"☆269Jun 20, 2022Updated 3 years ago
- The Official PyTorch Implementation of "LSGM: Score-based Generative Modeling in Latent Space" (NeurIPS 2021)☆369Dec 2, 2021Updated 4 years ago
- Masked Diffusion Transformer is the SOTA for image synthesis. (ICCV 2023)☆594Apr 23, 2024Updated last year
- Minimal implementation of scalable rectified flow transformers, based on SD3's approach☆635Jul 1, 2024Updated last year
- PyTorch Implementation of Diffusion Schrodinger Bridge Matching☆204Feb 19, 2025Updated last year
- Consistency Distilled Diff VAE☆2,213Nov 7, 2023Updated 2 years ago
- Official PyTorch implementation of the paper: Flow Matching in Latent Space☆337Jan 20, 2025Updated last year
- ☆7,318Jul 2, 2024Updated last year
- [NeurIPS 2023] Official Implementation: "Consistent Diffusion Models"☆58Apr 17, 2023Updated 2 years ago
- Official Code for Paper "Think While You Generate: Discrete Diffusion with Planned Denoising" [ICLR 2025]☆85Apr 24, 2025Updated 10 months ago
- Karras et al. (2022) diffusion models for PyTorch☆2,575Feb 12, 2026Updated last month
- [CVPR 2025 Oral] Improving Diffusion Inverse Problem Solving with Decoupled Noise Annealing☆177Dec 11, 2025Updated 3 months ago
- [ICML 2024 Best Paper] Discrete Diffusion Modeling by Estimating the Ratios of the Data Distribution (https://arxiv.org/abs/2310.16834)☆716Feb 29, 2024Updated 2 years ago
- A PyTorch library for implementing flow matching algorithms, featuring continuous and discrete flow matching implementations. It includes…☆4,246Jan 5, 2026Updated 2 months ago