mirror of
https://github.com/mudler/LocalAI.git
synced 2026-08-01 02:49:51 -04:00
Four defects the whole-branch review found on the Go side, plus two comment corrections. sample_rate is a disk-exhaustion hazard. The branch added the `form:` tag that makes the field bind for the first time, so the resample path went from dead to live, and utils.AudioResample interpolates the int straight into ffmpeg's -ar with no bound. Measured with ffmpeg 7: -ar 999999999 on a 0.01 s clip writes 20 MB and exits 0, which scales linearly to the reported 3.9 GB for one second, into a GeneratedContentDir nothing sweeps, and convertStems repeats it once per separation stem. Clamped to 8000..192000 in the handler, before the temp dir and before the model is touched, and rejected with a 400 outside it. The low end was reported as "a 0-byte file". It is not: -ar 1 writes a 78-byte header with no audio behind it, whose declared data size still claims 70 bytes, so go-audio parses it as a 35 SECOND file and a size check does not see it. The guard therefore compares the declared data chunk against the bytes actually on disk, and AudioResample now fails rather than returning a WAV carrying nothing. Both parts of a transform request land in one temp dir, and the raw copy was named only after the client's basename, so `-F audio=@mic/clip.wav -F reference=@loopback/clip.wav` wrote "raw-clip.wav" twice. Since AudioToWavPreservingShape hardlinks an already-PCM16 WAV rather than copying it, the reference part's os.Create truncated the inode audio.wav pointed at: mic and reference came out identical, which makes an echo canceller null everything and return near-silence with a 200. The raw copy now carries the form field name. audio-cpp had no BackendCapabilities entry, so VoiceCloningForModel returned nil before it ever consulted the model's tts.voice_cloning override and every `voice: "profile:<id>"` request was refused with a 400, on a backend that ships audio-cpp-chatterbox whose family serves cloning and not plain TTS. Registered with its RPCs, usecases and the reference-audio contract, and deliberately without the 16 kHz mono fold, which its separation families cannot survive. GetBackendCapability was exact-match only, so every pinned gallery variant read as an unknown backend: vulkan-localvqe lost the 16 kHz mono fold that used to be unconditional and started failing inside LocalVQE, and the usecase gate does not stand in for it because BuildFilteredFirstAvailableDefaultModel returns early once the client names a model. Lookup now falls back to the meta name by stripping the gallery's hardware prefix and release-channel suffix, exact match first so nothing can be shadowed. Same class as #10945. Also corrected: the AudioTransformRequest comment claimed echo's binder falls back to the field name, which it does not in either direction (bindData binds ONLY tagged fields and `continue`s otherwise; `model` arrives from setModelNameFromRequest's c.FormValue). And the stable_audio `src` heap corruption caveat now lives on ElevenLabsSoundGenerationRequest, where the Go developer who would add the field can see it, instead of only in C++. Assisted-by: Claude:claude-opus-5 [Claude Code] Signed-off-by: Ettore Di Giacinto <mudler@localai.io>
235 lines
9.9 KiB
Markdown
235 lines
9.9 KiB
Markdown
+++
|
||
disableToc = false
|
||
title = "Audio Transform"
|
||
weight = 34
|
||
url = "/features/audio-transform/"
|
||
+++
|
||
|
||

|
||
|
||
The audio-transform endpoints take **audio in** and emit **audio out**, optionally
|
||
conditioned on a second reference audio signal. The category is generic by
|
||
design - concrete operations include joint **acoustic echo cancellation +
|
||
noise suppression + dereverberation** (LocalVQE), voice conversion (reference
|
||
= target speaker), pitch shifting, audio super-resolution, and so on.
|
||
|
||
The first shipping backend is [LocalVQE](https://github.com/localai-org/LocalVQE),
|
||
a 1.3 M-parameter GGML-based model that performs joint AEC + noise suppression
|
||
+ dereverberation on 16 kHz mono speech, ~9.6× realtime on a desktop CPU. It
|
||
is a derivative of the Microsoft DeepVQE paper.
|
||
|
||
Source separation and voice conversion are served by the
|
||
[audio.cpp backend]({{%relref "features/audio-cpp" %}}): its `htdemucs` and
|
||
`mel_band_roformer` families produce the named stems described below, and
|
||
`seed_vc`, `vevo2` and `miocodec` do voice conversion against a reference
|
||
speaker.
|
||
|
||
## The mental model
|
||
|
||
Every audio-transform request carries:
|
||
|
||
- **`audio`** - the primary input file (required).
|
||
- **`reference`** - an auxiliary signal whose meaning is backend-specific (optional).
|
||
- For echo cancellation: the loopback / far-end signal played through the speakers.
|
||
- For voice conversion: the target speaker's reference clip.
|
||
- For pitch / style transfer: a tonal or style reference.
|
||
- When omitted, the backend treats it as silence and degrades gracefully (LocalVQE,
|
||
for example, does denoise + dereverb only when ref is empty).
|
||
- **`params`** - a generic `key=value` map forwarded to the backend.
|
||
- LocalVQE keys: `noise_gate=true|false`, `noise_gate_threshold_dbfs=<float>`.
|
||
|
||
This shape mirrors WebRTC's `ProcessStream(near)` / `ProcessReverseStream(far)`
|
||
APM API, NVIDIA Maxine's `NvAFX_Run` paired-stream signature, and the ICASSP
|
||
AEC challenge 2-channel WAV convention.
|
||
|
||
## Batch endpoint
|
||
|
||
`POST /audio/transformations` (alias `POST /audio/transform`) - multipart
|
||
form-data, returns audio bytes.
|
||
|
||
| Field | Type | Required | Notes |
|
||
|---|---|---|---|
|
||
| `model` | string | yes | Audio-transform model id (e.g. `localvqe-v1.3-4.8m`) |
|
||
| `audio` | file | yes | Primary input audio |
|
||
| `reference` | file | no | Optional auxiliary signal |
|
||
| `response_format` | string | no | `wav` (default), `mp3`, `ogg`, `flac` |
|
||
| `sample_rate` | int | no | Desired output sample rate in Hz. Omit it for the backend's own rate; otherwise it must be between 8000 and 192000, and a value outside that range is refused with a 400 |
|
||
| `params[<key>]` | string | no | Repeated; forwarded to backend |
|
||
| `params[stem]` | string | no | Multi-output transforms only; picks which named output the body carries (see [stems](#multi-output-transforms-source-separation-stems)) |
|
||
|
||
First install an audio-transform model from the gallery (the examples below use `localvqe-v1.3-4.8m`):
|
||
|
||
```bash
|
||
local-ai run localvqe-v1.3-4.8m
|
||
```
|
||
|
||
Example (LocalVQE: cancel echo, suppress noise, gate residual):
|
||
|
||
```bash
|
||
curl -X POST http://localhost:8080/audio/transformations \
|
||
-F model=localvqe-v1.3-4.8m \
|
||
-F audio=@mic.wav \
|
||
-F reference=@loopback.wav \
|
||
-F 'params[noise_gate]=true' \
|
||
-F 'params[noise_gate_threshold_dbfs]=-50' \
|
||
-o enhanced.wav
|
||
```
|
||
|
||
When `reference` is omitted, LocalVQE zero-fills the reference channel and
|
||
the operation reduces to noise suppression + dereverberation.
|
||
|
||
### What LocalAI does to your upload before the backend sees it
|
||
|
||
By default, **nothing**: the file reaches the backend at its own sample rate and
|
||
its own channel count. A WAV already carrying plain 16-bit PCM is passed through
|
||
byte for byte; any other container or encoding is transcoded to 16-bit PCM WAV
|
||
with the rate and the channel layout kept.
|
||
|
||
The exception is a backend that declares it needs a fixed input shape.
|
||
**LocalVQE** does: its echo cancellation is trained on 16 kHz mono and needs the
|
||
primary input and the reference in the same shape, so uploads for it are folded
|
||
to 16 kHz mono s16 with ffmpeg. The declaration is
|
||
`BackendCapability.AudioTransformInputMono16k` in `core/config`, and `localvqe`
|
||
is currently the only backend that sets it.
|
||
|
||
This matters for anything that is not speech enhancement. Source separation
|
||
models refuse any sample rate but their checkpoint's own (44.1 kHz for every
|
||
published htdemucs and mel_band_roformer checkpoint) and rely on the stereo
|
||
image to tell a centred vocal from a wide mix, so a 16 kHz mono downmix would
|
||
remove both the format they accept and the cue they work from.
|
||
|
||
### Multi-output transforms (source separation stems)
|
||
|
||
Some transforms produce **several** named outputs from one run: htdemucs yields
|
||
`drums`, `bass`, `other` and `vocals` in a single pass. The response body can
|
||
carry only one file, so:
|
||
|
||
- The backend runs **once** and writes every stem beside the main output.
|
||
- `params[stem]=<name>` chooses which one the body carries. Without it the
|
||
default is `vocals` when the model has one, and the model's first output
|
||
otherwise. An unknown stem name is refused with an error listing the real
|
||
ones, never silently substituted.
|
||
- Every stem, including the one in the body, is named in the **`X-Audio-Stems`**
|
||
response header, a compact JSON array:
|
||
|
||
```
|
||
X-Audio-Stems: [{"name":"drums","url":"/generated-audio/transform.drums.wav"},
|
||
{"name":"bass","url":"/generated-audio/transform.bass.wav"},
|
||
{"name":"other","url":"/generated-audio/transform.other.wav"},
|
||
{"name":"vocals","url":"/generated-audio/transform.vocals.wav"}]
|
||
```
|
||
|
||
Fetch any of those URLs to get the other stems without paying for a second
|
||
separation. `sample_rate` and `response_format` are applied to the stems as well
|
||
as to the body, so the whole set stays in the shape you asked for. The header is
|
||
listed in `Access-Control-Expose-Headers`, so browser clients can read it.
|
||
|
||
Single-output transforms (echo cancellation, voice conversion) do not set the
|
||
header at all, and `params[stem]` against such a model is refused rather than
|
||
ignored.
|
||
|
||
```bash
|
||
# isolate the vocals (the default), then see where the other stems went
|
||
curl -sS -D headers.txt -X POST http://localhost:8080/audio/transformations \
|
||
-F model=htdemucs -F audio=@song.wav -o vocals.wav
|
||
grep -i '^x-audio-stems' headers.txt
|
||
|
||
# or ask for a specific stem in the body
|
||
curl -sS -X POST http://localhost:8080/audio/transformations \
|
||
-F model=htdemucs -F audio=@song.wav -F 'params[stem]=drums' -o drums.wav
|
||
```
|
||
|
||
The stems live in the generated-content directory beside the main output and are
|
||
served from `/generated-audio/`. Like every other generated artifact, they are
|
||
not swept automatically.
|
||
|
||
## Streaming endpoint
|
||
|
||
`GET /audio/transformations/stream` - bidirectional WebSocket. The first
|
||
client message is a JSON envelope; subsequent client messages are binary
|
||
PCM frames; server emits binary PCM frames at the same cadence.
|
||
|
||
### Wire format
|
||
|
||
**Client → server** (text frame, first):
|
||
|
||
```json
|
||
{
|
||
"type": "session.update",
|
||
"model": "localvqe-v1.3-4.8m",
|
||
"sample_format": "S16_LE",
|
||
"sample_rate": 16000,
|
||
"frame_samples": 256,
|
||
"params": { "noise_gate": "true" }
|
||
}
|
||
```
|
||
|
||
`sample_format` is `S16_LE` (16-bit signed little-endian) or `F32_LE` (32-bit
|
||
float little-endian, [-1, 1]). `frame_samples` defaults to the backend's
|
||
preferred hop length (256 = 16 ms for LocalVQE).
|
||
|
||
**Client → server** (binary frames, subsequent): interleaved stereo PCM,
|
||
channel 0 = audio (mic), channel 1 = reference. Frame size:
|
||
`frame_samples × 2 channels × sample_size`. For `S16_LE` at 256 samples that
|
||
is 1024 bytes per frame; for `F32_LE` it is 2048 bytes. If the reference is
|
||
silent (no auxiliary signal), send zeros on channel 1.
|
||
|
||
**Server → client** (binary frames): mono PCM in the same format,
|
||
`frame_samples × sample_size` bytes (512 bytes for `S16_LE`, 1024 for `F32_LE`).
|
||
|
||
**Mid-stream control** (text frame): another `session.update` resets the
|
||
streaming state when its `reset` field is true; a `session.close` text frame
|
||
ends the session cleanly.
|
||
|
||
### Latency
|
||
|
||
LocalVQE has 16 ms algorithmic latency (one hop). At runtime the per-frame CPU
|
||
cost depends on the model: ~1.6 ms for the compact 1.3 M models (v1.1/v1.2,
|
||
~9.7× realtime) and ~3.3 ms for the wider v1.3 4.8 M model (~4.7× realtime) on
|
||
a 4-thread modern desktop, leaving the rest of the budget for network and
|
||
downstream playback.
|
||
|
||
## Backend-specific tuning (LocalVQE)
|
||
|
||
| `params[<key>]` | Type | Default | Effect |
|
||
|---|---|---|---|
|
||
| `noise_gate` | bool | `false` | Enable post-OLA RMS-based residual-echo gate |
|
||
| `noise_gate_threshold_dbfs` | float | `-45.0` | Gate threshold in dBFS; frames below are zeroed |
|
||
|
||
The gate is most useful in far-end-only / silent-near-end stretches where the
|
||
model's residual would otherwise sound like buffering or amplified noise floor.
|
||
A reasonable starting point is `-50` dBFS.
|
||
|
||
## Configuring a model
|
||
|
||
LocalVQE ships several weight releases in the gallery: `localvqe-v1.3-4.8m`
|
||
(current default - best quality), `localvqe-v1.2-1.3m` and `localvqe-v1.1-1.3m`
|
||
(compact, ~¼ the per-hop cost - good for low-core or power-constrained hosts).
|
||
All share the same backend and request API; only the `model` filename differs.
|
||
|
||
```yaml
|
||
name: localvqe
|
||
backend: localvqe
|
||
parameters:
|
||
model: localvqe-v1.3-4.8M-f32.gguf
|
||
|
||
# Backend-specific defaults can be set in Options[]; per-request
|
||
# params[*] form fields override.
|
||
#
|
||
# `backend` and `device` route through the upstream localvqe options
|
||
# builder so you can force a non-default GGML backend (e.g. `Vulkan`) or
|
||
# pin to a specific GPU index. Leave both unset to keep the CPU default.
|
||
options:
|
||
- noise_gate=true
|
||
- noise_gate_threshold_dbfs=-50
|
||
# - backend=Vulkan
|
||
# - device=0
|
||
```
|
||
|
||
## See also
|
||
|
||
- [Text to Audio (TTS)]({{< relref "text-to-audio.md" >}})
|
||
- [Audio to Text]({{< relref "audio-to-text.md" >}})
|
||
- [LocalVQE upstream](https://github.com/localai-org/LocalVQE)
|
||
- [DeepVQE paper (Indenbom et al., Interspeech 2023)](https://arxiv.org/abs/2306.03177)
|