Text-to-Image
Diffusers
StableDiffusionPipeline
stable-diffusion
stable-diffusion-diffusers

Is it possible to use it on an Inpainting task?

#1
by Bikesuffer - opened

I am wondering is it possible to use the unet to replace the unet of the inpainting task?

Segmind org

Yeah, Our Model is compatible with the Pipelines of the diffusers module, You can simply put our Model Card (segmind/small-sd) while loading and it will work. For Details on how to use an Inpainting Pipeline, Please refer to this guide, https://huggingface.co/docs/diffusers/using-diffusers/inpaint

Warlord-K changed discussion status to closed

Actually I test small-sd for inpainting. But the result seems not correct(the generated area is not related to the non-masked area). I am wondering is there a way that I can train the small-sd unet to work on inpainting task.

Sign up or log in to comment