c5ab99e74f
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.
106 lines
4.1 KiB
YAML
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"
|