jannerm / ddpo
Code for the paper "Training Diffusion Models with Reinforcement Learning"
☆380Updated 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☆464Updated 9 months ago
- ☆320Updated last month
- Code for "Diffusion Model Alignment Using Direct Preference Optimization"☆308Updated this week
- AlignProp uses direct reward backpropogation for the alignment of large-scale text-to-image diffusion models. Our method is 25x more samp…☆254Updated 2 months ago
- [CVPR 2024] Code for the paper "Using Human Feedback to Fine-tune Diffusion Models without Any Reward Model"☆185Updated 9 months ago
- Implementation of a single layer of the MMDiT, proposed in Stable Diffusion 3, in Pytorch☆285Updated this week
- Code for Fast Training of Diffusion Models with Masked Transformers☆385Updated 8 months ago
- ☆211Updated last year
- Implementation of MagViT2 Tokenizer in Pytorch☆588Updated this week
- Implementation of rectified flow and some of its followup research / improvements in Pytorch☆231Updated this week
- A collection of resources and papers on Vector Quantized Variational Autoencoder (VQ-VAE) and its application☆241Updated 9 months ago
- 🚀 PyTorch Implementation of "Progressive Distillation for Fast Sampling of Diffusion Models(v-diffusion)"☆225Updated 2 years ago
- code for "Diffusion Forcing: Next-token Prediction Meets Full-Sequence Diffusion"☆684Updated last month
- Consistency Models Made Easy☆253Updated 3 months ago
- This repo contains the code for 1D tokenizer and generator☆645Updated this week
- Implementation of Autoregressive Diffusion in Pytorch☆343Updated 2 months ago
- [ICCV 2023] Efficient Diffusion Training via Min-SNR Weighting Strategy☆238Updated last month
- Official PyTorch implementation of Video Probabilistic Diffusion Models in Projected Latent Space (CVPR 2023).☆314Updated 8 months ago
- MoVQGAN - model for the image encoding and reconstruction☆212Updated last year
- Official PyTorch Implementation of "SiT: Exploring Flow and Diffusion-based Generative Models with Scalable Interpolant Transformers"☆742Updated 10 months ago
- ☆258Updated 3 months ago
- Official pytorch implementation of the paper: "An Edit Friendly DDPM Noise Space: Inversion and Manipulations". CVPR 2024.☆305Updated 6 months ago
- A mini-library for training consistency models.☆234Updated last year
- Masked Diffusion Transformer is the SOTA for image synthesis. (ICCV 2023)☆545Updated 8 months ago
- ☆115Updated last year
- Rectified Diffusion: Straightness Is Not Your Need☆161Updated last month
- Official Pytorch Implementation of Representation Alignment for Generation: Training Diffusion Transformers Is Easier Than You Think☆791Updated last month
- Smooth Diffusion: Crafting Smooth Latent Spaces in Diffusion Models arXiv 2023 / CVPR 2024☆329Updated 3 months ago
- Scalable Diffusion Models with State Space Backbone☆149Updated 10 months ago
- SpeeD: A Closer Look at Time Steps is Worthy of Triple Speed-Up for Diffusion Model Training☆162Updated 2 months ago