jannerm / ddpo
Code for the paper "Training Diffusion Models with Reinforcement Learning"
☆414Updated 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☆511Updated last year
- ☆411Updated 3 months ago
- [CVPR 2024] Code for the paper "Using Human Feedback to Fine-tune Diffusion Models without Any Reward Model"☆206Updated 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…☆278Updated 4 months ago
- code for "Diffusion Forcing: Next-token Prediction Meets Full-Sequence Diffusion"☆777Updated last week
- Code for "Diffusion Model Alignment Using Direct Preference Optimization"☆397Updated last month
- Code for Fast Training of Diffusion Models with Masked Transformers☆396Updated 10 months ago
- [ICLR 2025] Rectified Diffusion: Straightness Is Not Your Need☆201Updated last week
- ☆275Updated 5 months ago
- 🚀 PyTorch Implementation of "Progressive Distillation for Fast Sampling of Diffusion Models(v-diffusion)"☆234Updated 2 years ago
- Implementation of Autoregressive Diffusion in Pytorch☆364Updated 4 months ago
- Consistency Models Made Easy☆276Updated 5 months ago
- ☆211Updated last year
- Smooth Diffusion: Crafting Smooth Latent Spaces in Diffusion Models arXiv 2023 / CVPR 2024☆333Updated 5 months ago
- Implementation of rectified flow and some of its followup research / improvements in Pytorch☆263Updated last month
- Official PyTorch Implementation of "SiT: Exploring Flow and Diffusion-based Generative Models with Scalable Interpolant Transformers"☆790Updated last year
- This repo contains the code for 1D tokenizer and generator☆748Updated last week
- Implementation of a single layer of the MMDiT, proposed in Stable Diffusion 3, in Pytorch☆328Updated 2 months ago
- Minimal Implementation of a D3PM in pytorch☆205Updated 11 months ago
- [ICCV 2023] Efficient Diffusion Training via Min-SNR Weighting Strategy☆243Updated 3 months ago
- EDM2 and Autoguidance -- Official PyTorch implementation☆663Updated 3 months ago
- MoVQGAN - model for the image encoding and reconstruction☆223Updated last year
- Implementation of MagViT2 Tokenizer in Pytorch☆597Updated 2 months ago
- [CVPR'23] Video Probabilistic Diffusion Models in Projected Latent Space☆314Updated 10 months ago
- [ICLR'25 Oral] Representation Alignment for Generation: Training Diffusion Transformers Is Easier Than You Think☆882Updated last week
- A mini-library for training consistency models.☆238Updated last year
- The collection of awesome papers on alignment of diffusion models.☆141Updated 2 weeks ago
- Annotated Flow Matching paper☆176Updated 6 months ago
- Official PyTorch implementation of the paper "In-Context Learning Unlocked for Diffusion Models"☆398Updated 11 months ago
- [ICLR2024] The official implementation of paper "VDT: General-purpose Video Diffusion Transformers via Mask Modeling", by Haoyu Lu, Guoxi…☆230Updated 10 months ago