justinpinkney / stable-diffusion

MIT License
1.45k stars 266 forks source link

Datasets used in fine-tuning for inpainting #21

Closed cats-food closed 2 years ago

cats-food commented 2 years ago

Huge thanks for your code contrubution first! Looks like the inpainting model is fine-tuned on Liaon dataset, the provided script laion.py seems doesn't work, could you please provide the exact link so that i can access the dataset (or its subset)? Also, if the pretrained model weights are available, could you please share it? Thanks in advance : )

justinpinkney commented 2 years ago

I haven't tried training an inpainting model. I think that code was leftover from the original repo (pesser/stable-diffusion)