This is an implementation of the segmind/SSD-1B 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 lora_url=@trained_model.tar -i prompt="a photo of TOK"
"a photo of TOK"