Home

Awesome

Stable Diffusion Cog model

This is an implementation of the Diffusers Stable Diffusion 1.4 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="monkey scuba diving"

Or, build a Docker image:

cog build

Or, push it to Replicate:

cog push r8.im/...