riffusion-inference/integrations
Hayk Martiros 93d90b3832 revup virtual diff target
2fc9b8d379
d98efa55fe6e64c65f3344a22cad8db1111240ff
450a9b6fad
a7b41d90bd
2023-01-17 02:17:23 +00:00
..
README.md revup virtual diff target 2023-01-17 02:17:23 +00:00
__init__.py revup virtual diff target 2023-01-17 02:17:23 +00:00
baseten.py revup virtual diff target 2023-01-17 02:17:23 +00:00
cog_riffusion.py revup virtual diff target 2023-01-17 02:17:23 +00: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