diff --git a/README.md b/README.md index 4e65182..f77295b 100644 --- a/README.md +++ b/README.md @@ -53,3 +53,8 @@ If you want to run the LLaVA prompt-image alignment experiments, you need to ded +## Training using 🤗 `trl` + +🤗 `trl` provides a [`DDPOTrainer` class](https://huggingface.co/docs/trl/ddpo_trainer) which lets you fine-tune Stable Diffusion on different reward functions using DDPO. The integration supports LoRA, too. You can check out the [supplementary blog post](https://huggingface.co/blog/trl-ddpo) for additional guidance. The DDPO integration was contributed by @metric-space to `trl`. + +