yuanzhi-zhu / mini_edmLinks
Minimum implementation of EDM (Elucidating the Design Space of Diffusion-Based Generative Models) on cifar10 and mnist
☆57Updated last year
Alternatives and similar repositories for mini_edm
Users that are interested in mini_edm are comparing it to the libraries listed below
Sorting:
- Implementation of "Analyzing and Improving the Training Dynamics of Diffusion Models"☆95Updated last year
- Consistency Models Made Easy☆310Updated 11 months ago
- The codebase of our paper "Improving the Training of Rectified Flows", NeurIPS 2024☆123Updated 11 months ago
- Official PyTorch implementation for the paper Minimizing Trajectory Curvature of ODE-based Generative Models, ICML 2023☆88Updated 7 months ago
- Official code for "Maximum Likelihood Training for Score-Based Diffusion ODEs by High-Order Denoising Score Matching" (ICML 2022)☆63Updated 3 years ago
- Official Implementation for "Consistency Flow Matching: Defining Straight Flows with Velocity Consistency"☆240Updated 8 months ago
- PyTorch Implementation of Diffusion Schrodinger Bridge Matching☆177Updated 6 months ago
- ☆300Updated 11 months ago
- The official repository for the paper "Optimal Flow Matching: Learning Straight Trajectories in Just One Step" (NeurIPS 2024)☆91Updated 8 months ago
- PyTorch code and model checkpoints for Score identity Distillation (SiD) and its adversarial version (SiDA)☆133Updated 5 months ago
- A mini-library for training consistency models.☆247Updated last year
- Official code for "Improved Techniques for Maximum Likelihood Estimation for Diffusion ODEs" (ICML 2023)☆18Updated 6 months ago
- Implementation of Consistency Models (Song et al 2023) for few-step image generation in Jax.☆19Updated 2 years ago
- Official code for "DPM-Solver-v3: Improved Diffusion ODE Solver with Empirical Model Statistics" (NeurIPS 2023)☆123Updated last year
- code for "Generalized Schrödinger Bridge Matching" (ICLR 2024).☆79Updated last year
- Official Pytorch repository for "Diffusion Posterior Sampling for Linear Inverse Problem Solving: A Filtering Perspective", where FPS (Fi…☆47Updated last year
- ☆278Updated 9 months ago
- [ICLR 2023]DEIS: Fast Sampling of Diffusion Models with Exponential Integrator☆158Updated 2 years ago
- Implementation of rectified flow and some of its followup research / improvements in Pytorch☆376Updated last month
- [ICML 2023] Reflected Diffusion Models (https://arxiv.org/abs/2304.04740)☆157Updated last year
- Official implementation for Estimating the Optimal Covariance with Imperfect Mean in Diffusion Probabilistic Models (ICML 2022), and a re…☆109Updated 3 years ago
- Educational implementation of the Discrete Flow Matching paper☆114Updated last year
- Official implementation of our paper "Bidirectional Consistency Models"; and reproduced Improved Consistency Models (iCT).☆20Updated 4 months ago
- Code for the paper Analytic-DPM: an Analytic Estimate of the Optimal Reverse Variance in Diffusion Probabilistic Models (ICLR 2022 Outsta…☆174Updated 3 years ago
- Pytorch implementation of "Light Schrödinger Bridge" (ICLR 2024)☆59Updated last year
- official code for Diff-Instruct algorithm for one-step diffusion distillation☆78Updated 8 months ago
- [ICLR 2024] Official code for the paper 'Elucidating the Exposure Bias in Diffusion Models'☆47Updated 3 months ago
- [ICML 2024] Improving Diffusion Models for Inverse Problems Using Optimal Posterior Covariance☆54Updated 11 months ago
- Reward fine-tuning for Stable Diffusion models based on stochastic optimal control, including Adjoint Matching☆43Updated 3 months ago
- Code release for the paper Generative Modeling With Inverse Heat Dissipation☆76Updated 2 years ago