DDPO를 사용한 강화 학습

🤗 TRL 라이브러리와 🤗 Diffusers를 사용하여 보상 함수에 대한 강화 학습으로 Stable Diffusion을 파인튜닝할 수 있습니다. 이는 Black et al.이 Training Diffusion Models with Reinforcement Learning에서 소개한 Denoising Diffusion Policy Optimization(DDPO) 알고리즘을 통해 수행되며, 🤗 TRL에는 DDPOTrainer로 구현되어 있습니다.

자세한 내용은 DDPOTrainer API 참고 문서를 참고하고 TRL을 통한 DDPO로 Stable Diffusion 모델 파인튜닝 블로그 게시물을 확인하세요.

Update on GitHub