andreasjansson / cog-stable-diffusion-inpainting

Inpainting using RunwayML's stable-diffusion-inpainting checkpoint
Apache License 2.0
19 stars 9 forks source link

Stable Diffusion Inpainting Cog model

Replicate

This is an implementation of the Stable Diffusion Inpainting as a Cog model. Cog packages machine learning models as standard containers.

First, download the pre-trained weights with your Hugging Face auth token:

cog run script/download-weights <your-hugging-face-auth-token>

Then, you can run predictions:

cog predict -i prompt="a herd of grazing sheep" -i image=@desktop.png -i mask=@desktop-mask.png

To push a new version you need to be @andreasjansson. If you are, then run

./test_and_push.sh