Fine-tuning for my own dataset

boschresearch / ALDM

Official implementation of "Adversarial Supervision Makes Layout-to-Image Diffusion Models Thrive" (ICLR 2024)

https://yumengli007.github.io/ALDM/

GNU Affero General Public License v3.0

52 stars 3 forks source link

Fine-tuning for my own dataset #2

Closed marcello-sousa closed 9 months ago

marcello-sousa commented 10 months ago

Can I do fine-tuning for my own dataset just like with ControlNet?

YumengLi007 commented 10 months ago

Hi, yes you can. You will need to adapt the dataloader and semantic classes manually for your own dataset. The dataloader codes are set under ALDM/dataloader

marcello-sousa commented 9 months ago

Thank's!!