text2pod/docker-compose.yml at main · makr-code/text2pod · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
# docker-compose.yml — universal across cpu / cuda / rocm / xpu (+ jetson),
# on both amd64 and arm64. Pick the accelerator with --profile.
#
#   docker compose                       --profile cpu    up
#   DEVICE_TAG=cu130    docker compose   --profile cuda   up
#   DEVICE_TAG=jetson51 docker compose   --profile jetson up
#   DEVICE_TAG=rocm     docker compose   --profile rocm   up
#   DEVICE_TAG=xpu      docker compose   --profile xpu    up
#
# Requires Compose V2 (the `docker compose` plugin), NOT the legacy
# `docker-compose` v1 — only V2 honors deploy.resources.devices outside Swarm.
#
# Architecture (amd64 vs arm64) is NOT a profile: it is the host you run on plus
# the matching DEVICE_TAG image. The device wiring below is arch-agnostic.
#
# Confirm a profile expands (and that extends merged) before running it:
#   DEVICE_TAG=rocm docker compose --profile rocm config

services:
  # --- shared base: exists only for extends, never runs (dummy _base profile)
  ebook2audiobook-base:
    image: athomasson2/ebook2audiobook:${DEVICE_TAG:-cpu}
    build:
      context: .
      dockerfile: Dockerfile
      args:
        APP_VERSION: ${APP_VERSION:-26.7.16}
        DEVICE_TAG: ${DEVICE_TAG:-cpu}
    working_dir: /app
    entrypoint: ["bash", "ebook2audiobook.command", "--script_mode", "full_docker"]
    tty: true
    stdin_open: true
    ports:
      - "7860:7860"
    volumes:
      - ./ebooks:/app/ebooks:rw
      - ./audiobooks:/app/audiobooks:rw
      - ./models:/app/models:rw
      - ./voices:/app/voices:rw
      - ./tmp:/app/tmp:rw
    restart: unless-stopped
    profiles: [_base]

  # --- CPU: no device wiring at all -----------------------------------------
  ebook2audiobook-cpu:
    extends:
      service: ebook2audiobook-base
    profiles: [cpu]

  # --- NVIDIA discrete (amd64 dGPU, or arm64 SBSA) --------------------------
  # Needs the nvidia-container-toolkit installed on the host.
  ebook2audiobook-cuda:
    extends:
      service: ebook2audiobook-base
    profiles: [cuda]
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: all
              capabilities: [gpu]

  # --- NVIDIA Jetson / Tegra (arm64 L4T) ------------------------------------
  # Uses the nvidia runtime (set "default-runtime": "nvidia" in
  # /etc/docker/daemon.json, or rely on this runtime: line). The deploy.devices
  # path above does NOT work on Tegra — Jetson goes through the runtime.
  ebook2audiobook-jetson:
    extends:
      service: ebook2audiobook-base
    profiles: [jetson]
    runtime: nvidia
    environment:
      NVIDIA_VISIBLE_DEVICES: all
      NVIDIA_DRIVER_CAPABILITIES: all
    group_add:
      - video

  # --- AMD ROCm -------------------------------------------------------------
  ebook2audiobook-rocm:
    extends:
      service: ebook2audiobook-base
    profiles: [rocm]
    devices:
      - /dev/kfd
      - /dev/dri
    group_add:
      - video
      - render

  # --- Intel XPU (oneAPI / IPEX) --------------------------------------------
  ebook2audiobook-xpu:
    extends:
      service: ebook2audiobook-base
    profiles: [xpu]
    devices:
      - /dev/dri
    group_add:
      - render