Diffusers documentation
Reinforcement learning training with DDPO
Get started
Inference
Using diffusion pipelines
Adapters and personalization
Optimize and scale
Basic performance
Inference
Community methods
Quantization
Hardware-specific acceleration
Serving and scaling
Modular Diffusers
Train and fine-tune
OverviewCreate a dataset for trainingAdapt a model to a new taskTrain a diffusion model
Model recipes
Training methods
Textual InversionDreamBoothLoRACustom DiffusionLatent Consistency DistillationReinforcement learning training with DDPO
NeMo AutomodelResources
Contribute
API
You are viewing main version, which requires installation from source. If you'd like
regular pip install, checkout the latest stable version (v0.40.0).
Reinforcement learning training with DDPO
You can fine-tune Stable Diffusion on a reward function via reinforcement learning with the 🤗 TRL library and 🤗 Diffusers. This is done with the Denoising Diffusion Policy Optimization (DDPO) algorithm introduced by Black et al. in Training Diffusion Models with Reinforcement Learning, which is implemented in 🤗 TRL with the DDPOTrainer.
For more information, check out the DDPOTrainer API reference and the Finetune Stable Diffusion Models with DDPO via TRL blog post.