Go to file

float-trip e813c12852 .		2022-10-01 15:10:01 +00:00
celeryconfig.py	.	2022-10-01 15:10:01 +00:00
client.py	.	2022-10-01 15:10:01 +00:00
readme.md	.	2022-10-01 15:10:01 +00:00
requirements.txt	.	2022-10-01 15:10:01 +00:00
tasks.py	.	2022-10-01 15:10:01 +00:00
utils.py	.	2022-10-01 15:10:01 +00:00

readme.md

Marseygen

Stable Diffusion bot with distributed inference.

Usage

Set up InvokeAI on the gen workers and activate the ldm environment
Install rabbitmq and redis, add URLs to celeryconfig.py
git clone https://github.com/float-trip/marseygen
pip install -r marseygen/requirements.txt
mv marseygen/*.py InvokeAI && cd InvokeAI
- Running the gen workers from this dir circumvents some Python import issues that I don't care to figure out right now
Start the API worker

celery -A tasks worker -B --concurrency 1 --loglevel=INFO

Start a gen worker for each GPU

export CUDA_VISIBLE_DEVICES=0,
export WORKER_HOST="user@gen_worker_ip"
export WORKER_SSH_PORT="22"
export WORKER_ID="unique_id"
celery -A tasks worker -Q gen -n unique_name -B --concurrency 1 --loglevel=INFO`