jannerm / ddpo
Code for the paper "Training Diffusion Models with Reinforcement Learning"
☆450Updated 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☆553Updated last year
- AlignProp uses direct reward backpropogation for the alignment of large-scale text-to-image diffusion models. Our method is 25x more samp…☆282Updated 6 months ago
- ☆481Updated 5 months ago
- Code for "Diffusion Model Alignment Using Direct Preference Optimization"☆472Updated 3 months ago
- [CVPR 2024] Code for the paper "Using Human Feedback to Fine-tune Diffusion Models without Any Reward Model"☆222Updated last year
- Code for Fast Training of Diffusion Models with Masked Transformers☆401Updated 11 months ago
- 🚀 PyTorch Implementation of "Progressive Distillation for Fast Sampling of Diffusion Models(v-diffusion)"☆239Updated 2 years ago
- code for "Diffusion Forcing: Next-token Prediction Meets Full-Sequence Diffusion"☆843Updated last month
- Official PyTorch Implementation of "SiT: Exploring Flow and Diffusion-based Generative Models with Scalable Interpolant Transformers"☆826Updated last year
- ☆462Updated 2 years ago
- Masked Diffusion Transformer is the SOTA for image synthesis. (ICCV 2023)☆562Updated last year
- Consistency Models Made Easy☆277Updated 6 months ago
- ☆282Updated 6 months ago
- A collection of resources and papers on Vector Quantized Variational Autoencoder (VQ-VAE) and its application☆277Updated 3 months ago
- Implementation of Autoregressive Diffusion in Pytorch☆374Updated 6 months ago
- [ICLR 2025] Rectified Diffusion: Straightness Is Not Your Need☆216Updated last month
- EDM2 and Autoguidance -- Official PyTorch implementation☆699Updated 4 months ago
- The collection of awesome papers on alignment of diffusion models.☆200Updated last week
- Official Implementation of Rectified Flow (ICLR2023 Spotlight)☆1,217Updated 9 months ago
- An open-source toolbox for fast sampling of diffusion models. Official implementations of our works published in ICML, NeurIPS, CVPR.☆281Updated last month
- [ICCV 2023] Efficient Diffusion Training via Min-SNR Weighting Strategy☆247Updated 4 months ago
- Implementation of a single layer of the MMDiT, proposed in Stable Diffusion 3, in Pytorch☆349Updated 3 months ago
- ☆115Updated 2 years ago
- Unofficial PyTorch Implementation of Denoising Diffusion Probabilistic Models (DDPM)☆213Updated 8 months ago
- [ICLR'25 Oral] Representation Alignment for Generation: Training Diffusion Transformers Is Easier Than You Think☆986Updated last month
- [CVPR'23] Video Probabilistic Diffusion Models in Projected Latent Space☆315Updated 11 months ago
- [ICLR2024] The official implementation of paper "VDT: General-purpose Video Diffusion Transformers via Mask Modeling", by Haoyu Lu, Guoxi…☆237Updated last year
- Implementation of MagViT2 Tokenizer in Pytorch☆601Updated 3 months ago
- Implementation of rectified flow and some of its followup research / improvements in Pytorch☆285Updated last week
- Official Jax Implementation of MaskGIT☆509Updated 2 years ago