latent-diffusion icon indicating copy to clipboard operation
latent-diffusion copied to clipboard

Details about training inpainting model

Open LigZhong opened this issue 3 years ago • 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?

LigZhong avatar Jan 26 '22 07:01 LigZhong

The same question as @LigZhong.

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

roimulia2 avatar Jan 26 '22 14:01 roimulia2

+1 How did you train the inpainting model?

zzw-zwzhang avatar Mar 20 '22 15:03 zzw-zwzhang

+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)

david-svitov avatar Apr 27 '22 10:04 david-svitov

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

shensongli avatar Apr 24 '23 09:04 shensongli

The code and steps of training inpainting model is here

CreamyLong avatar Dec 28 '23 14:12 CreamyLong