Support for inpainting training for dreambooth?

XavierXiao / Dreambooth-Stable-Diffusion

Implementation of Dreambooth (https://arxiv.org/abs/2208.12242) with Stable Diffusion

MIT License

7.57k stars 790 forks source link

Support for inpainting training for dreambooth? #161

Open atharva-akridata opened 6 months ago

atharva-akridata commented 6 months ago

Hi, I see there's a file called "inpaint.py" under the scripts directory. Can I use the dreambooth training script to train an inpainting model and generate inpainted images? Or is this repo only for regular dreambooth and not inpainting?