CompVis/latent-diffusion

Details about training inpainting model

Opened this issue · 5 comments

Hi, this is an awesome work. I am looking into the code but cannot find the entrance to train the inpainting model. Can you elaborate it a little bit?

The same question as @LigZhong.

Do you happen to know if they have a pre-trained model for the inpainting?

+1
How did you train the inpainting model?

+1 How did you train the inpainting model?

It looks like you can use this config (https://github.com/CompVis/latent-diffusion/blob/main/models/ldm/inpainting_big/config.yaml) modifying it by adding 'data' section like in this config (https://github.com/CompVis/latent-diffusion/blob/main/configs/latent-diffusion/ffhq-ldm-vq-4.yaml).
But first you need to prepare the data like in this repo: https://github.com/saic-mdal/lama
I'm not quite sure it will work, but at least you can try it)

hello,how did you train the inpainting model?Could you give me some advice

The code and steps of training inpainting model is here