riffusion-inference/integrations
daanelson 2695af9706 cog using common load fxn 2023-01-05 00:01:55 -08:00
..
README.md cog using common load fxn 2023-01-05 00:01:55 -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:

cog push r8.im/hmartiro/riffusion