This is an implementation of a Cog model. Cog packages machine learning models as standard containers.
First, download the pre-trained weights:
cog run script/download-weights
Then, you can run
text2img predictions:
cog predict -i prompt="a moss covered astronaut with a black background"
img2img predictions:
cog predict -i [email protected] -i prompt="photo of a lone astronaut standing on a barren planet"
inpainting predictions:
cog predict -i [email protected] -i [email protected] -i prompt="a tabby cat, high resolution, sitting on a park bench"