An open API service indexing awesome lists of open source software.

https://github.com/saladtechnologies/qr-code-worker

This is a queue worker for generating stable diffusion qr codes with any of a variety of stable diffusion backends.
https://github.com/saladtechnologies/qr-code-worker

Last synced: 8 months ago
JSON representation

This is a queue worker for generating stable diffusion qr codes with any of a variety of stable diffusion backends.

Awesome Lists containing this project

README

          

# qr-code-worker
This is a queue worker for generating stable diffusion qr codes with any of a variety of stable diffusion backends.

Currently Supported Backends:
- [x] [stable-fast-qr-code](https://github.com/SaladTechnologies/stable-fast-qr-demo)
- [x] [a1111](https://github.com/SaladTechnologies/a1111-dynamic)
- [x] [sdnext](https://github.com/SaladTechnologies/sdnext-dynamic)
- [x] [comfy](https://github.com/SaladTechnologies/comfyui-dynamic)
- [ ] invoke

## Build
```bash
# Usage: ./build --backend --base-image
./build --backend stable-fast-qr-code --base-image saladtechnologies/stable-fast-qr-code:0.4.0-baked
```

## Run
```bash
docker compose up
```

## Pre-Built Images

Image Names follow the format: `saladtechnologies/:worker-`.
For example, if you build worker v0.1.0 (per `package.json`) on top of `saladtechnologies/stable-fast-qr-code:0.4.0-baked`, the image name would be:
```
saladtechnologies/stable-fast-qr-code:worker0.1.0-0.4.0-baked
```

Images with models included are using:
- Checkpoint: [Dreamshaper 8](https://civitai.com/models/4384?modelVersionId=128713)
- Controlnet: [QR Code Monster v2.0](https://civitai.com/models/111006?modelVersionId=122143)

You should be able to use any stable diffusion 1.5 checkpoint, and any text2img QR code controlnet. Each backend supports dynamically loading models from civitai via environment variables, though the syntax may vary. Please refer to the documentation for each backend, linked above.

### Current Images

| Image Name | Base Image | Worker Version | Backend | Models Baked In |
| ---------- | ---------- | -------------- | ------- | --------------- |
| `saladtechnologies/stable-fast-qr-code:worker0.1.0-0.5.0-baked` | `saladtechnologies/stable-fast-qr-code:0.5.0-baked` | `0.1.0` | `stable-fast-qr-code` | yes |
| `saladtechnologies/stable-fast-qr-code:worker0.1.0-0.5.0` | `saladtechnologies/stable-fast-qr-code:0.5.0` | `0.1.0` | `stable-fast-qr-code` | no |
| `saladtechnologies/sdnext:worker0.1.0-122143-128713` | `saladtechnologies/sdnext:122143-128713` | `0.1.0` | `sdnext` | yes |
| `saladtechnologies/sdnext:worker0.1.0-dynamic` | `saladtechnologies/sdnext:dynamic` | `0.1.0` | `sdnext` | no |
| `saladtechnologies/a1111:worker0.1.0-122143-128713` | `saladtechnologies/a1111:122143-128713` | `0.1.0` | `a1111` | yes |
| `saladtechnologies/a1111:worker0.1.0-dynamic` | `saladtechnologies/a1111:dynamic` | `0.1.0` | `a1111` | no |
| `saladtechnologies/comfyui:worker0.1.0-baked` | `saladtechnologies/comfyui:baked` | `0.1.0` | `comfy` | yes |
| `saladtechnologies/comfyui:worker0.1.0-dynamic` | `saladtechnologies/comfyui:dynamic` | `0.1.0` | `comfy` | no |

## Environment Variables

| Name | Description | Default |
| ---- | ----------- | ------- |
| QUEUE_NAME | The name of the queue to get jobs from | **REQUIRED** |
| QUEUE_URL | The URL of the queue to get jobs from | **REQUIRED** |
| QUEUE_API_KEY | The API key for the queue | **REQUIRED** |
| BENCHMARK_ID | The ID of the benchmark to report results to | **REQUIRED** |
| REPORTING_URL | The URL to report results to | **REQUIRED** |
| REPORTING_API_KEY | The API key for the reporting endpoint | **REQUIRED** |
| IMAGE_SIZE | The size of the qr image to generate | `512` |
| STARTUP_CHECK_INTERVAL | The interval in ms to check for the image generation service liveness | `1000` |
| STARTUP_CHECK_MAX_TRIES | The maximum number of times to check for the image generation service liveness | `300` |
| EAGER_NUMBER | This queue worker uses eager fetching. This is the number of jobs to keep locally pre-fetched | `2` |