Dedicated Inpainting Model

[jaidevshriram]

Hey! Thanks for making this a thing! Is there any plan to release a dedicated inpainting model (where the diffusion model conditions on the mask / masked image)? The zero-shot technique shown currently only supports smaller masks, so the dedicated model would be very useful. This model could be similar to Imagen Editor - https://arxiv.org/abs/2212.06909