Buckets:

rtrm's picture
|
download
raw
615 Bytes

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.

Xet Storage Details

Size:
615 Bytes
·
Xet hash:
ee46958395fa4962dd9585bc630feb27eb05d778838e2d334fc9f7e3707ced32

Xet efficiently stores files, intelligently splitting them into unique chunks and accelerating uploads and downloads. More info.