jannerm / ddpo
Code for the paper "Training Diffusion Models with Reinforcement Learning"
☆406Updated last year
Alternatives and similar repositories for ddpo:
Users that are interested in ddpo are comparing it to the libraries listed below
- DDPO for finetuning diffusion models, implemented in PyTorch with LoRA support☆496Updated 11 months ago
- [CVPR 2024] Code for the paper "Using Human Feedback to Fine-tune Diffusion Models without Any Reward Model"☆203Updated 11 months ago
- AlignProp uses direct reward backpropogation for the alignment of large-scale text-to-image diffusion models. Our method is 25x more samp…☆276Updated 4 months ago
- Code for "Diffusion Model Alignment Using Direct Preference Optimization"☆383Updated last month
- ☆405Updated 3 months ago
- Consistency Models Made Easy☆272Updated 5 months ago
- Code for Fast Training of Diffusion Models with Masked Transformers☆396Updated 9 months ago
- 🚀 PyTorch Implementation of "Progressive Distillation for Fast Sampling of Diffusion Models(v-diffusion)"☆234Updated 2 years ago
- [ICLR 2025] Rectified Diffusion: Straightness Is Not Your Need☆195Updated this week
- Smooth Diffusion: Crafting Smooth Latent Spaces in Diffusion Models arXiv 2023 / CVPR 2024☆332Updated 5 months ago
- A collection of resources and papers on Vector Quantized Variational Autoencoder (VQ-VAE) and its application☆260Updated last month
- Implementation of a single layer of the MMDiT, proposed in Stable Diffusion 3, in Pytorch☆321Updated 2 months ago
- Implementation of rectified flow and some of its followup research / improvements in Pytorch☆257Updated last month
- [ICCV 2023] Efficient Diffusion Training via Min-SNR Weighting Strategy☆241Updated 3 months ago
- ☆211Updated last year
- An open-source toolbox for fast sampling of diffusion models. Official implementations of our works published in ICML, NeurIPS, CVPR.☆258Updated last week
- ☆271Updated 5 months ago
- Official pytorch implementation of the paper: "An Edit Friendly DDPM Noise Space: Inversion and Manipulations". CVPR 2024.☆323Updated 8 months ago
- MoVQGAN - model for the image encoding and reconstruction☆221Updated last year
- Official PyTorch implementation of the paper "In-Context Learning Unlocked for Diffusion Models"☆394Updated 11 months ago
- code for "Diffusion Forcing: Next-token Prediction Meets Full-Sequence Diffusion"☆766Updated last week
- EDM2 and Autoguidance -- Official PyTorch implementation☆653Updated 3 months ago
- [ICLR2025] Halton Scheduler for Masked Generative Image Transformer☆196Updated 2 weeks ago
- Official PyTorch implementation of Video Probabilistic Diffusion Models in Projected Latent Space (CVPR 2023).☆314Updated 10 months ago
- Masked Diffusion Transformer is the SOTA for image synthesis. (ICCV 2023)☆552Updated 10 months ago
- Implementation of Autoregressive Diffusion in Pytorch☆360Updated 4 months ago
- This repo contains the code for 1D tokenizer and generator☆706Updated 2 weeks ago
- ☆115Updated 2 years ago
- Unofficial Implementation of Consistency Models in pytorch☆255Updated 2 years ago
- PyTorch implementation for "Parallel Sampling of Diffusion Models", NeurIPS 2023 Spotlight☆132Updated last year