History

Dan 1dd4dbeb03 updating cog readme		2023-01-19 22:05:02 -08:00
..
README.md	updating cog readme	2023-01-19 22:05:02 -08:00
__init__.py	Greatly simplify the server and baseten integration	2022-12-26 17:47:11 -08:00
baseten.py	Enable ruff import sorting	2022-12-26 18:12:02 -08:00
cog_riffusion.py	cog using common load fxn	2023-01-05 00:01:55 -08:00

README.md

Integrations

This package contains integrations of Riffusion into third party apps and deployments.

Baseten

Baseten is a platform for building and deploying machine learning models.

Replicate

To run riffusion as a Cog model, first, install Cog and download the model weights:

cog run python -m integrations.cog_riffusion --download_weights

Then you can run predictions:

cog predict -i prompt_a="funky synth solo"

You can also view the model on Replicate here. Owners can push an updated version of the model like so:

# download weights locally if you haven't already
cog run python -m integrations.cog_riffusion --download_weights

cog login
cog push r8.im/riffusion/riffusion