riffusion icon indicating copy to clipboard operation
riffusion copied to clipboard

working cog predict, updated readme

Open daanelson opened this issue 3 years ago • 2 comments

Hey! Added a cog.yaml and predict.py file so that we have the code that powers the riffusion demo on Replicate (https://replicate.com/hmartiro/riffusion) colocated with the riffusion source inference code.

daanelson avatar Dec 19 '22 22:12 daanelson

Thanks @daanelson. In order to work with replicate, do these files have to live at the top level or can they be in a subdirectory?

hmartiro avatar Dec 20 '22 05:12 hmartiro

good q @hmartiro. predict.py can live in a subdirectory, but cog.yaml has to be in the project root. I went ahead and moved predict.py into the riffusion module & renamed it cog_riffusion.py, let me know if that works or if you have something else in mind.

daanelson avatar Dec 21 '22 00:12 daanelson

@daanelson is it possible to allow for custom seed image input to the cog? Or some of the other options that are available on the inference server like a negative prompt?

rhelsing avatar Dec 27 '22 17:12 rhelsing

@daanelson sorry for the delay. With the v0.3.0 release (https://github.com/riffusion/riffusion/releases/tag/v0.3.0) this should simplify the code and integration

hmartiro avatar Dec 27 '22 17:12 hmartiro

@daanelson since I didn't know how to push to your branch / this PR, I duplicated and updated it at #57. I'll tag you there for a question

hmartiro avatar Dec 29 '22 20:12 hmartiro