Open chavinlo opened 1 year ago
if you just want to use the inpainting model, you shouldn't have to do any training. Download the pretrained inpainting model and use the CLI commands in the readme.
if you really do want to train a custom inpaint model on your own dataset, I strongly recommend resuming from my checkpoint instead of the base SD model.
if you just want to use the inpainting model, you shouldn't have to do any training. Download the pretrained inpainting model and use the CLI commands in the readme.
if you really do want to train a custom inpaint model on your own dataset, I strongly recommend resuming from my checkpoint instead of the base SD model.
Thanks for the recommendation, yeah it was pretty silly training it from scratch, I was using the Waifu Diffusion model. Btw, how many steps and images have you trained your inpaint model on? required resolution still is 512x512 right?
I just added some code to load the weights from an uninitialized SD model, should work better now if you want to train from a base SD model.
I trained for 100k steps at 256 batch size with the LAION aesthetic dataset, which took about a week on 8xA100. I trained with 512 resolution but it's an adjustable parameter.
Hello, I was trying out the out/inpainting method you mentioned in the readme.
After training for about 1000 steps, I tried out sample.py with inpainting. Input Images:
Output:
![image](https://user-images.githubusercontent.com/85657083/193423087-22009149-1687-44d7-bcc0-cc781bc319e2.png)
I have also read from other issues that at least 10k steps are necesarry, is there something wrong I could be doing?