|
Download diffusers/docs/source/en/training/ddpo.md from 6chan/Cobra: direct link, hf CLI and curl.
- Browser
- Download file 1.21 kB
-
https://huggingface.co/spaces/6chan/Cobra/resolve/main/diffusers/docs/source/en/training/ddpo.md
- Command line
-
hf download hf://spaces/6chan/Cobra/diffusers/docs/source/en/training/ddpo.md
-
curl -L -o ddpo.md https://huggingface.co/spaces/6chan/Cobra/resolve/main/diffusers/docs/source/en/training/ddpo.md
1.21 kB
A newer version of the Gradio SDK is available: 6.31.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 [~trl.DDPOTrainer].
For more information, check out the [~trl.DDPOTrainer] API reference and the Finetune Stable Diffusion Models with DDPO via TRL blog post.