Official PyTorch implementation of RB-Modulation: Training-Free Personalization of Diffusion Models using Stochastic Optimal Control.
🔥 Code release: Official implementation has been released at https://github.com/google/RB-Modulation.
Given reference images of preferred style or content, our method, RB-Modulation, offers a plug-and-play solution for (a) stylization with various prompts, and (b) composition with reference content images while maintaining sample diversity and prompt alignment.
- [2024.08.23] RB-Modulation code is released!
- [2024.05.29] Paper is published on arXiv!
@article{rout2024rbmodulation,
title={RB-Modulation: Training-Free Personalization of Diffusion Models using Stochastic Optimal Control},
author={Litu Rout and Yujia Chen and Nataniel Ruiz and Abhishek Kumar and Constantine Caramanis and Sanjay Shakkottai and Wen-Sheng Chu},
journal={arXiv preprint arXiv:2405.17401},
year={2024}
}
Copyright © 2024, Google LLC. All rights reserved.