jannerm / ddpoLinks
Code for the paper "Training Diffusion Models with Reinforcement Learning"
☆551Updated 2 years ago
Alternatives and similar repositories for ddpo
Users that are interested in ddpo are comparing it to the libraries listed below
Sorting:
- DDPO for finetuning diffusion models, implemented in PyTorch with LoRA support☆740Updated last year
- ☆735Updated last year
- Code for "Diffusion Model Alignment Using Direct Preference Optimization"☆661Updated 3 months ago
- code for "Diffusion Forcing: Next-token Prediction Meets Full-Sequence Diffusion"☆1,152Updated 3 months ago
- [CVPR 2024] Code for the paper "Using Human Feedback to Fine-tune Diffusion Models without Any Reward Model"☆242Updated last year
- AlignProp uses direct reward backpropogation for the alignment of large-scale text-to-image diffusion models. Our method is 25x more samp…☆310Updated last year
- The collection of awesome papers on alignment of diffusion models. [ACM Computing Surveys]☆401Updated this week
- [ICLR 2025] Rectified Diffusion: Straightness Is Not Your Need☆245Updated 10 months ago
- A collection of resources and papers on Vector Quantized Variational Autoencoder (VQ-VAE) and its application☆329Updated last year
- [ICLR'25 Oral] Representation Alignment for Generation: Training Diffusion Transformers Is Easier Than You Think☆1,534Updated 10 months ago
- Implementation of Autoregressive Diffusion in Pytorch☆432Updated 2 months ago
- Consistency Models Made Easy☆324Updated last year
- JAX implementation of MeanFlow☆539Updated last week
- Official PyTorch Implementation of "SiT: Exploring Flow and Diffusion-based Generative Models with Scalable Interpolant Transformers"☆1,092Updated last month
- Code for Fast Training of Diffusion Models with Masked Transformers☆421Updated last year
- Implementation of a single layer of the MMDiT, proposed in Stable Diffusion 3, in Pytorch☆514Updated 3 weeks ago
- Implementation of MagViT2 Tokenizer in Pytorch☆661Updated last year
- Masked Diffusion Transformer is the SOTA for image synthesis. (ICCV 2023)☆594Updated last year
- [CVPR'23] Video Probabilistic Diffusion Models in Projected Latent Space☆324Updated last year
- [NeurIPS 2025] An official implementation of Flow-GRPO: Training Flow Matching Models via Online RL☆1,979Updated 3 months ago
- Scaling Diffusion Transformers with Mixture of Experts☆415Updated last year
- Implementation of rectified flow and some of its followup research / improvements in Pytorch☆424Updated last week
- ☆317Updated 2 months ago
- SEED-Voken: A Series of Powerful Visual Tokenizers☆992Updated 2 months ago
- An open-source toolbox for fast sampling of diffusion models. Official implementations of our works published in ICML, NeurIPS, CVPR, J. …☆355Updated 2 months ago
- EDM2 and Autoguidance -- Official PyTorch implementation☆816Updated last year
- [ICML 2025] Official PyTorch Implementation of "History-Guided Video Diffusion"☆613Updated 7 months ago
- [ICCV 2025] Official implementation of the paper: REPA-E: Unlocking VAE for End-to-End Tuning of Latent Diffusion Transformers☆449Updated 2 months ago
- [ICLR 2026 Oral] DiffusionNFT: Online Diffusion Reinforcement with Forward Process☆626Updated this week
- Official Jax Implementation of MaskGIT☆554Updated 3 years ago