This is an implementation of the amused/amused-512 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="a photo of a colorful lemon mocktail can with mint leaves with an abstract background"
"a photo of a colorful lemon mocktail can with mint leaves with an abstract background"