Gist that mostly just takes the notebook Stability AI have in the github repo, and turns it into a script you can run locally after installing a few dependencies from pip:

https://gist.github.com/cmsj/2d6b247ad4fc8f15011105feeda763e...