Files
esh-pfi-infrastructure/playbooks/deploy-kokoro.yaml
T
vh c5ab99e74f kokoro: persist custom voices across container recreate
Wrapper only enumerates one voice directory (settings.voices_dir,
default /app/api/src/voices/v1_0 — inside the container's writable
layer, not bind-mounted). Override via VOICES_DIR=/app/user_voices
(host bind mount) and add a command shim that cp -r's built-ins from
the in-image v1_0 into user_voices on every start. Built-ins re-seed
fresh from the image (so upgrades that add voices propagate); custom
.pt files in user_voices are preserved (cp -r is additive).

Also adds scripts/blend_kokoro_voice.py + a playbook around it that
mirrors the wrapper's request-time voice="a(w)+b(w)" math but writes
the result as a named .pt to user_voices, making it discoverable via
GET /v1/audio/voices and persistent across recreate. Defaults to
athena = af_bella(2)+af_aoede(1) normalized.
2026-05-10 17:48:24 -07:00

106 lines
4.1 KiB
YAML

# Deploy Kokoro-FastAPI to irv-ml1.
#
# Image is published on GHCR — no Dockerfile to maintain, no first-run
# model download (Kokoro-82M weights are baked in). Stage compose +
# .env, pull, bring up. ~6.5 GB pull on cold cache, ~2-5 min.
#
# Usage:
# scripts/elway irv-ml1 --playbook playbooks/deploy-kokoro.yaml
#
# Idempotent — every step is creates-/when-gated; rerun is safe.
vars:
compose_dir: /opt/docker/compose/kokoro
user_voices_dir: /worktank/kokoro/user_voices
voices_dir: /worktank/kokoro/voices
host_port: "8193"
steps:
# ── host-side dirs ──────────────────────────────────────────────────
- name: Ensure /worktank/kokoro root exists (one-time, sudo)
shell: mkdir -p /worktank/kokoro
sudo: true
creates: /worktank/kokoro
- name: Chown /worktank/kokoro to lkraven
shell: chown lkraven:lkraven /worktank/kokoro
sudo: true
when: '[ "$(stat -c %U /worktank/kokoro)" != lkraven ]'
- name: Ensure user-voices dir exists
shell: mkdir -p {{ user_voices_dir }}
creates: "{{ user_voices_dir }}"
- name: Chown user-voices to 1001:1001 (appuser inside container)
# The compose `command:` shim runs as appuser (image USER) and
# cp's the in-image built-in voicepacks into this dir on every
# container start. appuser is uid 1001 inside the container; the
# host has no user with that uid, so we chown numerically. Once
# set, the shim can write; the blend script (run via docker exec
# -u 0) writes as root, producing 0644 files appuser can read.
shell: chown 1001:1001 {{ user_voices_dir }}
sudo: true
when: '[ "$(stat -c %u:%g {{ user_voices_dir }})" != "1001:1001" ]'
- name: Ensure voices dir exists (used only if compose mount is enabled)
shell: mkdir -p {{ voices_dir }}
creates: "{{ voices_dir }}"
- name: Ensure compose dir exists
shell: mkdir -p {{ compose_dir }}
creates: "{{ compose_dir }}"
# ── deploy compose files ────────────────────────────────────────────
- name: Upload compose.yaml
upload:
src: stacks/kokoro/compose.yaml
dest: "{{ compose_dir }}/compose.yaml"
mode: "0644"
- name: Seed .env from template (only if absent)
upload:
src: stacks/kokoro/.env.example
dest: "{{ compose_dir }}/.env"
mode: "0644"
when: "[ ! -f {{ compose_dir }}/.env ]"
# ── pull + bring up ─────────────────────────────────────────────────
- name: "docker compose pull (first run: ~6.5 GB from GHCR)"
# --quiet suppresses pull-command progress, but the docker daemon
# still emits its own per-layer extraction progress events
# ("ffbfd7a09415 Extracting 64.06MB" etc) on stderr — these are
# what flooded the log on the previous run. Drop them with grep.
# set -o pipefail so a real pull failure isn't swallowed.
shell: |
set -o pipefail
cd {{ compose_dir }} && docker compose pull --quiet 2>&1 \
| grep -vE '^\s*[0-9a-f]{8,16}\s+(Extracting|Pull complete|Pulling fs layer|Verifying Checksum|Download complete|Waiting|Already exists|Downloading|Pulling from)'
- name: docker compose up -d
shell: cd {{ compose_dir }} && docker compose up -d
- name: Wait for /v1/audio/voices to respond
shell: |
for i in $(seq 1 60); do
curl -sf -o /dev/null --max-time 3 http://localhost:{{ host_port }}/v1/audio/voices && exit 0
sleep 5
done
exit 1
changed_when: "false"
verify:
- name: /v1/audio/voices returns 200
shell: curl -sf -o /dev/null http://localhost:{{ host_port }}/v1/audio/voices
changed_when: "false"
- name: /v1/audio/voices includes at least one built-in (af_bella)
shell: curl -sf http://localhost:{{ host_port }}/v1/audio/voices | grep -q af_bella
changed_when: "false"
- name: Container is running
shell: docker inspect kokoro --format '{{.State.Status}}' | grep -q running
changed_when: "false"