Files
esh-pfi-infrastructure/stacks/vibevoice/README.md
T
vh 4549d241a7 stacks: add Kokoro, VibeVoice 1.5B, Chatterbox Turbo (TTS slate fill-in)
Three TTS additions to round out coverage on irv-ml1, each filling a
distinct niche the existing slate doesn't own.

Final coverage matrix (all on irv-ml1):
  Kokoro              — low-latency English, fixed voice library, ~300ms TTFA
  Chatterbox Turbo    — low-latency English w/ voice cloning + paralinguistic tags
  IndexTTS-2          — English voice cloning + emotion vector / text control
  Qwen3-TTS-1.7B-Base — high-quality English voice cloning
  CosyVoice 3         — multilingual (Chinese-leaning)
  VibeVoice 1.5B      — long-form / multi-speaker dialogue

stacks/kokoro:
  - port 8193, GPU device 0 (3090)
  - pulls ghcr.io/remsky/kokoro-fastapi-gpu:v0.2.4-master (no Dockerfile,
    no first-run model download — models baked in)
  - 60+ built-in voices, OpenAI-compat with stream=true over chunked HTTP
  - Apache-2.0 weights + code, ~1 GB VRAM

stacks/vibevoice:
  - port 8194, GPU device 1 (A6000 — for 7B headroom)
  - builds groxaxo/VibeVoice-FastAPI1 (more current fork of ncoder-ai)
    pinned to 7614c469a145
  - default model microsoft/VibeVoice-1.5B (~7 GB bf16 VRAM); env var
    swap to rsxdalv/VibeVoice-Large (7B) or FabioSarracino/VibeVoice-Large-Q8
  - multi-speaker dialogue via /v1/vibevoice/generate with Speaker N: format
  - long-form niche only — not low-latency

stacks/chatterbox:
  - port 8196, GPU device 0 (3090)
  - builds devnen/Chatterbox-TTS-Server (most active Turbo-supporting wrapper)
  - default model ResembleAI/chatterbox-turbo (~2.5 GB fp16, ~75ms latency)
  - paralinguistic tags inline ([laugh] [whisper] etc) — different shape
    from IndexTTS-2's emotion vector; fills the speed+cloning niche
    Kokoro/IndexTTS don't cover together
  - mandatory PerTh watermark on outputs (Resemble policy)

Three matching playbooks under playbooks/deploy-{kokoro,vibevoice,
chatterbox}.yaml. All idempotent, creates-/when-gated.

Cold-deploy disk on /worktank/: ~7 GB Kokoro + ~19 GB VibeVoice 1.5B
+ ~12 GB Chatterbox = ~38 GB total. VRAM concurrent: ~10-11 GB across
both GPUs.

Skipped from the original four-stack proposal: VibeVoice Realtime
(overlaps Kokoro's niche; Kokoro wins on latency, license, and not
needing a build).
2026-04-25 16:18:37 -07:00

116 lines
4.1 KiB
Markdown

# VibeVoice
Microsoft's diffusion-based long-form TTS, served via
[groxaxo/VibeVoice-FastAPI1](https://github.com/groxaxo/VibeVoice-FastAPI1)
(a recent fork of [ncoder-ai/VibeVoice-FastAPI](https://github.com/ncoder-ai/VibeVoice-FastAPI)
which moves faster than upstream).
Model: [microsoft/VibeVoice-1.5B](https://huggingface.co/microsoft/VibeVoice-1.5B)
by default. Switch to the 7B variant via `.env` if you want the
bigger checkpoint.
## Why this stack exists
Long-form / podcast-quality TTS with native multi-speaker dialogue
support. Designed for one-shot generation of multi-minute scripts
where conversation flow matters. **Not** for low-latency single-line
synthesis — for that use Kokoro or Chatterbox Turbo.
| | use case |
|---|---|
| **VibeVoice 1.5B** | long-form, multi-speaker dialogue (this stack) |
| Kokoro | low-latency English, fixed voice library |
| Chatterbox Turbo | low-latency English w/ voice cloning |
| IndexTTS-2 | English voice cloning + emotion control |
| Qwen3-TTS-1.7B-Base | high-quality English voice cloning |
| CosyVoice 3 | multilingual (Chinese-leaning) |
## API
OpenAI-compat at `http://10.100.79.3:8194`:
```bash
# Single-speaker (OpenAI-style).
curl -fsS -X POST http://10.100.79.3:8194/v1/audio/speech \
-H 'Content-Type: application/json' \
-d '{"model":"vibevoice","input":"Hello there.","voice":"voice-name","response_format":"wav"}' \
> out.wav
# Multi-speaker dialogue — the headline feature. Format the input
# as a script with `Speaker N:` prefixes (0-indexed). The wrapper's
# extended /v1/vibevoice/generate endpoint handles voice switching.
curl -fsS -X POST http://10.100.79.3:8194/v1/vibevoice/generate \
-H 'Content-Type: application/json' \
-d '{
"script":"Speaker 0: Welcome to the show.\nSpeaker 1: Glad to be here.\nSpeaker 0: Today we discuss…",
"voices":["voice-host","voice-guest"],
"stream":true
}' > podcast.wav
# List available voices.
curl http://10.100.79.3:8194/v1/audio/voices
```
OpenAPI / docs at `/docs`. Healthcheck at `/health`.
`stream=true` is honored on the multi-speaker endpoint; the
single-shot OpenAI endpoint returns the full file in one go.
## Voices
Drop `.wav` / `.mp3` / `.flac` / `.m4a` into
`/worktank/vibevoice/voices/` on the host (mounted read-only into
the container). Restart the container after adding; the wrapper
scans the dir at init, not per-request:
```bash
ssh irv-ml1 'cd /opt/docker/compose/vibevoice && docker compose restart'
```
VibeVoice also has built-in voice presets (Carter, Davis, Emma,
Frank, Grace, Mike, Samuel) accessible by name. Microsoft has not
released the cloning tooling so you can't add new "trained" voices
— but the bundled ones already cover most podcast use cases.
## Deploy
```bash
scripts/elway irv-ml1 --playbook playbooks/deploy-vibevoice.yaml
```
Cold deploy budget:
- ~12 GB image build (CUDA 12.8 + torch 2.8 + flash-attn)
- ~7 GB model download (VibeVoice-1.5B) on first start
- **Total: ~19 GB on /worktank/vibevoice/**
First build: ~12 min. First generation: ~30-60 s warmup.
## Switching to the 7B variant
```bash
ssh irv-ml1 '
cd /opt/docker/compose/vibevoice
sed -i "s|^VIBEVOICE_MODEL=.*|VIBEVOICE_MODEL=rsxdalv/VibeVoice-Large|" .env
docker compose up -d
'
```
The 7B model auto-downloads on next start (~18 GB). VRAM jumps from
~7 GB to ~18 GB bf16 — keep `VIBEVOICE_GPU_DEVICES=1` (A6000) for it.
For lower VRAM at slight quality cost, set `VIBEVOICE_QUANT=int8_torchao`
which brings 7B down to ~10 GB.
## Gotchas
- **Not streaming-friendly for single-line use.** The diffusion head
has to denoise the whole latent before vocoding. Streaming on
`/v1/vibevoice/generate` works at script-segment granularity
(paragraph-ish), not token-by-token.
- **Voice cloning isn't published.** Microsoft released the inference
models but not the training pipeline. Use the built-in voices, or
pick another stack (IndexTTS-2 / Qwen3-TTS / Chatterbox Turbo).
- **`flash_attention_2`** is the upstream default; if your GPU/torch
combo doesn't have it built, set `VIBEVOICE_ATTN=sdpa` in `.env`
to fall back to PyTorch's scaled-dot-product attention.
- **License**: VibeVoice MIT (Microsoft); wrapper MIT.