There are two key differences between this and the stock Cog model:
- The Safety Detector is deactivated
- The service expects a callback URL where it will post every step of the diffusion
This is an implementation of the Diffusers Stable Diffusion v2.1 as 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 predictions:
cog predict -i prompt="monkey scuba diving"