jannerm / ddpo
Code for the paper "Training Diffusion Models with Reinforcement Learning"
☆340Updated last year
Related projects ⓘ
Alternatives and complementary repositories for ddpo
- DDPO for finetuning diffusion models, implemented in PyTorch with LoRA support☆427Updated 7 months ago
- Code for "Diffusion Model Alignment Using Direct Preference Optimization"☆261Updated 10 months ago
- AlignProp uses direct reward backpropogation for the alignment of large-scale text-to-image diffusion models. Our method is 25x more samp…☆241Updated last week
- [CVPR 2024] Code for the paper "Using Human Feedback to Fine-tune Diffusion Models without Any Reward Model"☆169Updated 7 months ago
- ☆164Updated 3 weeks ago
- Consistency Models Made Easy☆215Updated 3 weeks ago
- Smooth Diffusion: Crafting Smooth Latent Spaces in Diffusion Models arXiv 2023 / CVPR 2024☆317Updated last month
- Code for Fast Training of Diffusion Models with Masked Transformers☆370Updated 5 months ago
- [ICCV 2023] Efficient Diffusion Training via Min-SNR Weighting Strategy☆226Updated 6 months ago
- A mini-library for training consistency models.☆213Updated 10 months ago
- code for "Diffusion Forcing: Next-token Prediction Meets Full-Sequence Diffusion"☆588Updated 3 weeks ago
- ☆212Updated last year
- Official PyTorch implementation of Video Probabilistic Diffusion Models in Projected Latent Space (CVPR 2023).☆303Updated 5 months ago
- MoVQGAN - model for the image encoding and reconstruction☆184Updated last year
- Rectified Diffusion: Straightness Is Not Your Need☆117Updated last week
- ☆231Updated last month
- Implementation of Autoregressive Diffusion in Pytorch☆291Updated this week
- PyTorch implementation for "Parallel Sampling of Diffusion Models", NeurIPS 2023 Spotlight☆124Updated last year
- Implementation of a single layer of the MMDiT, proposed in Stable Diffusion 3, in Pytorch☆252Updated 2 months ago
- ☆108Updated last year
- Implementation of rectified flow and some of its followup research / improvements in Pytorch☆181Updated this week
- Analyzing and Improving the Training Dynamics of Diffusion Models (EDM2)☆521Updated 5 months ago
- An open-source toolbox for fast sampling of diffusion models. Official implementations of our works published in ICML, NeurIPS, CVPR.☆196Updated last month
- Official PyTorch implementation of the paper "In-Context Learning Unlocked for Diffusion Models"☆378Updated 7 months ago
- ☆111Updated last year
- This repo contains the code for 1D tokenizer and generator☆527Updated this week
- unofficial MaskGIT reproduction in PyTorch☆163Updated 9 months ago
- SpeeD: A Closer Look at Time Steps is Worthy of Triple Speed-Up for Diffusion Model Training☆160Updated 3 weeks ago
- Scalable Diffusion Models with State Space Backbone☆145Updated 8 months ago
- ☆454Updated last year