vh
81efa8da96
Upstream Zonos ships only Gradio + Python SDK — no REST surface — so
asset-engine (which routes a clean JSON POST to /v1/audio/speech) can't
target it directly. Add a thin FastAPI adapter (stacks/zonos/adapter/):
POST /v1/audio/speech in front of the Zonos SDK, built FROM local/zonos
to reuse torch/CUDA/SDK. Returns a JSON envelope {audio, audio_format,
seed} — the seed rides back so asset-engine regenerate/fork can pin it
(Zonos is the fleet's first genuinely seedable TTS). compose gains a
zonos-api service on 8201; .env.example gains the port + voices dir.
103 lines
4.4 KiB
YAML
103 lines
4.4 KiB
YAML
# Zonos-v0.1 (Zyphra's expressive multilingual open-weight TTS) served
|
|
# via the OFFICIAL Zyphra/Zonos repo's Gradio interface.
|
|
#
|
|
# Why this stack exists alongside the other TTS:
|
|
# * Apache-2.0, 44 kHz output, zero-shot voice cloning from a
|
|
# 10-30 s sample, and explicit emotion/conditioning sliders —
|
|
# a different control surface from tag-based engines.
|
|
# * Multilingual (EN/JA/ZH/FR/DE), trained on 200k+ hours.
|
|
# * Cheap to run (~6 GB VRAM, transformer variant ~3.6 GB weights).
|
|
#
|
|
# CAVEAT — this is an AUDITION/EVAL surface, not a skaldsong-pluggable
|
|
# engine yet: the official repo ships a Gradio WebUI + Python SDK, NOT
|
|
# an OpenAI-compatible /v1/audio/speech endpoint. To wire Zonos into
|
|
# skaldsong's router we'd need the community FastAPI fork (Zyphra/Zonos
|
|
# PR #73) or a thin adapter. Bench it by ear first; promote later if it
|
|
# earns a slot.
|
|
#
|
|
# Image is built locally from the upstream repo's Dockerfile via docker
|
|
# buildx git-context. All tunables live in .env — edit that, not this.
|
|
|
|
services:
|
|
zonos:
|
|
image: local/zonos:${ZONOS_TAG}
|
|
build:
|
|
# Single Dockerfile at repo root; bundles espeak-ng (required for
|
|
# Zonos's eSpeak phonemization).
|
|
context: https://github.com/Zyphra/Zonos.git#${ZONOS_SHA}
|
|
dockerfile: Dockerfile
|
|
container_name: zonos
|
|
restart: unless-stopped
|
|
runtime: nvidia
|
|
# Upstream's Dockerfile sets NO CMD — it launches the app from its own
|
|
# compose instead. Without this the NVIDIA entrypoint prints the CUDA
|
|
# banner and exits 0, looping forever (nothing binds 7860). Mirror
|
|
# upstream: python3 gradio_interface.py (demo.launch binds 0.0.0.0:7860).
|
|
command: ["python3", "gradio_interface.py"]
|
|
ports:
|
|
- "${ZONOS_BIND:-0.0.0.0}:${ZONOS_PORT}:7860"
|
|
environment:
|
|
- NVIDIA_VISIBLE_DEVICES=${ZONOS_GPU_DEVICES:-0}
|
|
- NVIDIA_DRIVER_CAPABILITIES=compute,utility
|
|
# Make Gradio bind all interfaces inside the container so the host
|
|
# port-map reaches it (Gradio otherwise may bind 127.0.0.1 only).
|
|
- GRADIO_SERVER_NAME=0.0.0.0
|
|
# Explicit: gradio_interface.py reads GRADIO_SHARE (defaults False).
|
|
# Pin it off so a missing default never tries a public share tunnel
|
|
# (irv-ml1 has no egress for that).
|
|
- GRADIO_SHARE=False
|
|
- HF_HOME=/app/hf_cache
|
|
volumes:
|
|
- ${ZONOS_CACHE_DIR}:/app/hf_cache
|
|
healthcheck:
|
|
# No JSON status route on the Gradio app — just probe the root
|
|
# page for liveness. start_period covers the first-boot model pull.
|
|
test: ["CMD-SHELL", "python3 -c \"import urllib.request,sys; urllib.request.urlopen('http://127.0.0.1:7860/', timeout=5); sys.exit(0)\""]
|
|
interval: 30s
|
|
timeout: 10s
|
|
retries: 3
|
|
start_period: 600s
|
|
labels:
|
|
- homepage.group=AI Systems
|
|
- homepage.name=Zonos
|
|
- homepage.icon=mdi-waveform
|
|
- homepage.description=Expressive multilingual TTS + cloning, 44kHz (Gradio eval, irv-ml1)
|
|
- homepage.href=http://10.100.79.3:${ZONOS_PORT}
|
|
|
|
# OpenAI-ish REST adapter (POST /v1/audio/speech) that asset-engine
|
|
# routes to — upstream Zonos has no REST surface, only Gradio + SDK.
|
|
# Built FROM the gradio image above (reuses torch/CUDA/SDK); loads its
|
|
# own copy of the model, so it adds ~6 GB VRAM on top of the gradio
|
|
# service. Drop the gradio service once Zonos earns a permanent slot.
|
|
zonos-api:
|
|
image: local/zonos-api:${ZONOS_TAG}
|
|
build:
|
|
context: ./adapter
|
|
dockerfile: Dockerfile
|
|
args:
|
|
ZONOS_BASE: local/zonos:${ZONOS_TAG}
|
|
container_name: zonos-api
|
|
restart: unless-stopped
|
|
runtime: nvidia
|
|
depends_on:
|
|
- zonos
|
|
ports:
|
|
- "${ZONOS_BIND:-0.0.0.0}:${ZONOS_API_PORT}:8000"
|
|
environment:
|
|
- NVIDIA_VISIBLE_DEVICES=${ZONOS_GPU_DEVICES:-0}
|
|
- NVIDIA_DRIVER_CAPABILITIES=compute,utility
|
|
- HF_HOME=/app/hf_cache
|
|
- ZONOS_MODEL=${ZONOS_MODEL:-Zyphra/Zonos-v0.1-transformer}
|
|
- ZONOS_VOICES_DIR=/app/voices
|
|
volumes:
|
|
- ${ZONOS_CACHE_DIR}:/app/hf_cache
|
|
# Reference clips for zero-shot cloning, selected by the `voice`
|
|
# request field (filename under this dir).
|
|
- ${ZONOS_VOICES_DIR}:/app/voices:ro
|
|
healthcheck:
|
|
test: ["CMD-SHELL", "python3 -c \"import urllib.request,sys; urllib.request.urlopen('http://127.0.0.1:8000/health', timeout=5); sys.exit(0)\""]
|
|
interval: 30s
|
|
timeout: 10s
|
|
retries: 3
|
|
start_period: 600s
|