replicate / cog

Containers for machine learning
https://cog.run
Apache License 2.0
8.09k stars 563 forks source link

Is Redis worker supposed to exit after completing a single job? #856

Open yekta opened 1 year ago

yekta commented 1 year ago

I've been trying to make Redis worker work with no success, unless I'm misunderstanding something. I'm using this example with my own cog image: https://github.com/replicate/cog-redis-example

I add a message to the queue, the worker picks it up. Logs shows that it processes it but then it exits with the message:

Shutting down worker: bye bye!

No error messages, just that. I'm wondering if this is the intended behaviour of a cog. I expected it to continue listening for messages but maybe I understood the worker wrong. If this is the intended behaviour, are there any ways to make it not exit after a single job completion?

yekta commented 1 year ago

When I try to use the example with this image it also quits after a single job, no reason given is to why: https://replicate.com/stability-ai/stable-diffusion/api

If I use this with this image, it doesn't quit after one job: https://replicate.com/nitrosocke/redshift-diffusion/api

I don't understand why it's quitting with one and not the other. It doesn't say why it does that, just says Shutting down worker: bye bye!