blob: c0f7ba1973668990f3b0aea53b73947340ad149b (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
|
services:
imggen:
build: .
image: imggen:local
user: "${IMGGEN_UID:-1000}:${IMGGEN_GID:-100}"
group_add:
- video # gid for /dev/dri/card* access as non-root
devices:
- /dev/dri
volumes:
- ${IMGGEN_OUT:-./out}:/out
- ${IMGGEN_CACHE:-./models}:/models
environment:
- HF_HOME=/models
- IMGGEN_MODEL=${IMGGEN_MODEL:-sdxl}
- IMGGEN_PORT=${IMGGEN_PORT:-8765}
ports:
- "127.0.0.1:${IMGGEN_PORT:-8765}:8765"
entrypoint: ["python3"]
command: ["server.py"]
|