aboutsummaryrefslogtreecommitdiffstats
path: root/compose.yaml
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"]