thu-ml / unidiffuser

Code and models for the paper "One Transformer Fits All Distributions in Multi-Modal Diffusion"
GNU Affero General Public License v3.0
1.37k stars 86 forks source link

How to finetune unidiffuser #5

Open dawei03896 opened 1 year ago

dawei03896 commented 1 year ago

thanks for your great works! by the way can unidiffuser be finetuned like dreambooth?

baofff commented 1 year ago

Yes. Many finetuing algorithms on transformers are already available, such as https://arxiv.org/abs/2110.04366