318 lines
13 KiB
Python
318 lines
13 KiB
Python
# SPDX-License-Identifier: Apache-2.0
|
|
# SPDX-FileCopyrightText: Copyright contributors to the vLLM-Omni project
|
|
"""End-to-end online tests for higgs-audio v3 against /v1/audio/speech.
|
|
|
|
Mirrors the higgs_audio_v2 test layout. Covers the plain-text-in / audio-out
|
|
happy path, a small concurrent burst (Stage 0 prefix caching has to coexist
|
|
with batching), the voice-clone path via ``ref_audio`` + ``ref_text``, the
|
|
upstream ``references[]`` cookbook alias added by
|
|
``normalize_references_alias`` in ``protocol/audio.py``, and the few
|
|
validator rejections that should remain 4xx.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import os
|
|
|
|
os.environ.setdefault("VLLM_WORKER_MULTIPROC_METHOD", "spawn")
|
|
os.environ.setdefault("VLLM_USE_DEEP_GEMM", "0")
|
|
os.environ.setdefault("VLLM_MOE_USE_DEEP_GEMM", "0")
|
|
|
|
import pytest
|
|
|
|
from tests.helpers.mark import hardware_test
|
|
from tests.helpers.media import get_asset_path
|
|
from tests.helpers.runtime import OmniServerParams
|
|
from tests.helpers.stage_config import get_deploy_config_path
|
|
|
|
MODEL = "bosonai/higgs-audio-v3-tts-4b"
|
|
STAGE_CONFIG = get_deploy_config_path("higgs_multimodal_qwen3.yaml")
|
|
SERVER_ARGS = ["--trust-remote-code", "--disable-log-stats"]
|
|
SERVER_ENV = {"VLLM_USE_DEEP_GEMM": "0", "VLLM_MOE_USE_DEEP_GEMM": "0"}
|
|
|
|
TEST_PARAMS = [
|
|
pytest.param(
|
|
OmniServerParams(
|
|
model=MODEL,
|
|
stage_config_path=STAGE_CONFIG,
|
|
server_args=SERVER_ARGS,
|
|
env_dict=SERVER_ENV,
|
|
),
|
|
id="higgs_audio_v3_plain_text",
|
|
)
|
|
]
|
|
|
|
DEFAULT_SPEECH_TIMEOUT_S = 180.0
|
|
# Floor for ~0.5 s of 24 kHz mono PCM_16: 24000 * 0.5 * 2 bytes ~= 24 KiB.
|
|
# A WAV header adds 44 bytes; conservative floor catches truncated /
|
|
# silence-only outputs without flagging short legitimate clips.
|
|
_MIN_AUDIO_BYTES = 20_000
|
|
|
|
# Reuse the shared TTS reference clip (clean ~5 s 24 kHz mono human speech)
|
|
# vendored under tests/assets/qwen3_tts/. Keeps a single WAV across TTS
|
|
# tests rather than duplicating asset bytes.
|
|
_REF_AUDIO_URL = get_asset_path("qwen3_tts/clone_2.wav", as_data_url=True)
|
|
_REF_TEXT = "Okay. Yeah. I resent you. I love you. I respect you. But you know what? You blew it! And thanks to you."
|
|
|
|
|
|
@pytest.mark.parametrize("omni_server", TEST_PARAMS, indirect=True)
|
|
class TestHiggsAudioV3OnlineHappyPath:
|
|
"""Plain-text -> audio happy paths against the live HTTP server."""
|
|
|
|
@pytest.mark.slow
|
|
@pytest.mark.tts
|
|
@hardware_test(res={"cuda": "L4", "rocm": "MI325"}, num_cards=1)
|
|
def test_plain_text_wav(self, omni_server, online_client) -> None:
|
|
"""Single non-streaming WAV request - canonical TTS happy path."""
|
|
online_client.send_audio_speech_request(
|
|
{
|
|
"model": omni_server.model,
|
|
"input": "Hello world.",
|
|
"stream": False,
|
|
"response_format": "wav",
|
|
"timeout": DEFAULT_SPEECH_TIMEOUT_S,
|
|
# whisper-small mishears this short clip ~0.5% of the time; on a
|
|
# failed match re-verify with this stronger ASR before failing so
|
|
# the gate is not flaky. Local-only key (not forwarded to server).
|
|
"transcript_escalation_model": "large-v3",
|
|
}
|
|
)
|
|
|
|
@pytest.mark.slow
|
|
@pytest.mark.tts
|
|
@hardware_test(res={"cuda": "L4", "rocm": "MI325"}, num_cards=1)
|
|
def test_plain_text_with_max_new_tokens(self, omni_server, online_client) -> None:
|
|
"""``max_new_tokens`` is one of the few extra fields the v3 validator accepts."""
|
|
online_client.send_audio_speech_request(
|
|
{
|
|
"model": omni_server.model,
|
|
"input": "Innovation distinguishes between a leader and a follower.",
|
|
"stream": False,
|
|
"response_format": "wav",
|
|
"timeout": DEFAULT_SPEECH_TIMEOUT_S,
|
|
"max_new_tokens": 1024,
|
|
"min_audio_bytes": _MIN_AUDIO_BYTES,
|
|
}
|
|
)
|
|
|
|
@pytest.mark.slow
|
|
@pytest.mark.tts
|
|
@hardware_test(res={"cuda": "L4", "rocm": "MI325"}, num_cards=1)
|
|
def test_concurrent_plain_text(self, omni_server, online_client) -> None:
|
|
"""Three concurrent non-streaming requests - guards per-slot audio state and Stage-0 PC under batching."""
|
|
online_client.send_audio_speech_request(
|
|
{
|
|
"model": omni_server.model,
|
|
"input": "it was the night before my birthday",
|
|
"stream": False,
|
|
"response_format": "wav",
|
|
"timeout": DEFAULT_SPEECH_TIMEOUT_S,
|
|
},
|
|
request_num=3,
|
|
)
|
|
|
|
@pytest.mark.slow
|
|
@pytest.mark.tts
|
|
@hardware_test(res={"cuda": "L4", "rocm": "MI325"}, num_cards=1)
|
|
def test_plain_text_pcm_streaming(self, omni_server, online_client) -> None:
|
|
"""Streaming PCM via talker2code2wav_async_chunk + the connector's codec_streaming path.
|
|
|
|
The deploy YAML pins ``async_chunk: true`` and ``codec_streaming: true`` so chunks
|
|
flow Stage 0 -> Stage 1 per AR loop. The Stage-1 codec honors
|
|
``meta.left_context_size`` and ``meta.right_holdback_size`` so per-chunk codec
|
|
windows stitch into a coherent PCM stream. The byte-count gate is the same as
|
|
the sync paths; per-chunk audio content is verified offline against Whisper.
|
|
|
|
NOTE: ``min_hnr_db=0.0`` sit below the typical speech-noise floor so the
|
|
check still catches catastrophic codec failure (silence, white noise, sample
|
|
scramble all give HNR << 0) while allowing for the sliding-window codec's
|
|
slightly-noisier-than-sync output. The default 1.0 dB threshold has only
|
|
~0.16 dB margin over measured single-request output (1.16 dB on L4), so
|
|
flake risk is too high there.
|
|
"""
|
|
online_client.send_audio_speech_request(
|
|
{
|
|
"model": omni_server.model,
|
|
"input": "The quick brown fox jumps over the lazy dog.",
|
|
"stream": True,
|
|
"stream_format": "audio",
|
|
"response_format": "pcm",
|
|
"timeout": DEFAULT_SPEECH_TIMEOUT_S,
|
|
"min_hnr_db": 0.0,
|
|
}
|
|
)
|
|
|
|
@pytest.mark.slow
|
|
@pytest.mark.tts
|
|
@hardware_test(res={"cuda": "L4", "rocm": "MI325"}, num_cards=1)
|
|
def test_concurrent_pcm_streaming(self, omni_server, online_client) -> None:
|
|
"""Three concurrent streaming requests - guards per-request frame cursors
|
|
in ``talker2code2wav_async_chunk`` and per-slot delay-pattern state under batched AR.
|
|
|
|
NOTE: ``min_hnr_db=-2.0`` is looser than the single-request streaming test
|
|
because batched AR adds per-request codec quality variance (DAC boundary
|
|
artifacts compounded across the 3-way batch); -2.0 dB gives ~1.6 dB margin
|
|
over the measured worst-of-3 on L4 (-0.42 dB) while still well above the
|
|
catastrophic-failure region.
|
|
"""
|
|
online_client.send_audio_speech_request(
|
|
{
|
|
"model": omni_server.model,
|
|
"input": "She sells seashells by the seashore.",
|
|
"stream": True,
|
|
"stream_format": "audio",
|
|
"response_format": "pcm",
|
|
"timeout": DEFAULT_SPEECH_TIMEOUT_S,
|
|
"min_hnr_db": -2.0,
|
|
},
|
|
request_num=3,
|
|
)
|
|
|
|
|
|
@pytest.mark.parametrize("omni_server", TEST_PARAMS, indirect=True)
|
|
class TestHiggsAudioV3OnlineInlineControlTokens:
|
|
"""Inline control-token surface from the bosonai cookbook.
|
|
|
|
The model exposes four categories of ``<|category:value|>`` tokens
|
|
(emotion, style, prosody, sfx). Delivery tokens (emotion / style /
|
|
prosody speed-pitch-expressive) go at the START of input; positional
|
|
tokens (pause, sfx) go inline. SFX tokens must pair with their written
|
|
onomatopoeia.
|
|
|
|
These tests check the serving surface and spoken text. They do not assert
|
|
whether the requested control changes audio quality.
|
|
"""
|
|
|
|
@pytest.mark.slow
|
|
@pytest.mark.tts
|
|
@hardware_test(res={"cuda": "L4", "rocm": "MI325"}, num_cards=1)
|
|
def test_inline_emotion_and_expressive(self, omni_server, online_client) -> None:
|
|
"""Delivery tokens (emotion + expressive_high) at the start of input."""
|
|
online_client.send_audio_speech_request(
|
|
{
|
|
"model": omni_server.model,
|
|
"input": ("<|emotion:amusement|><|prosody:expressive_high|>Wait, that was actually hilarious."),
|
|
"stream": False,
|
|
"response_format": "wav",
|
|
"timeout": DEFAULT_SPEECH_TIMEOUT_S,
|
|
"min_audio_bytes": _MIN_AUDIO_BYTES,
|
|
"transcript_expected_text": "Wait, that was actually hilarious.",
|
|
}
|
|
)
|
|
|
|
@pytest.mark.slow
|
|
@pytest.mark.tts
|
|
@hardware_test(res={"cuda": "L4", "rocm": "MI325"}, num_cards=1)
|
|
def test_inline_style_whispering(self, omni_server, online_client) -> None:
|
|
"""Style token at the start - ``whispering``."""
|
|
online_client.send_audio_speech_request(
|
|
{
|
|
"model": omni_server.model,
|
|
"input": "<|style:whispering|>It is just between you and me, alright?",
|
|
"stream": False,
|
|
"response_format": "wav",
|
|
"timeout": DEFAULT_SPEECH_TIMEOUT_S,
|
|
"min_audio_bytes": _MIN_AUDIO_BYTES,
|
|
"transcript_expected_text": "It is just between you and me, alright?",
|
|
}
|
|
)
|
|
|
|
@pytest.mark.slow
|
|
@pytest.mark.tts
|
|
@hardware_test(res={"cuda": "L4", "rocm": "MI325"}, num_cards=1)
|
|
def test_inline_prosody_speed_and_pitch(self, omni_server, online_client) -> None:
|
|
"""Two prosody tokens at the start - slow speed plus low pitch."""
|
|
online_client.send_audio_speech_request(
|
|
{
|
|
"model": omni_server.model,
|
|
"input": (
|
|
"<|prosody:speed_slow|><|prosody:pitch_low|>The radar shows a storm approaching from the east."
|
|
),
|
|
"stream": False,
|
|
"response_format": "wav",
|
|
"timeout": DEFAULT_SPEECH_TIMEOUT_S,
|
|
"min_audio_bytes": _MIN_AUDIO_BYTES,
|
|
"transcript_expected_text": "The radar shows a storm approaching from the east.",
|
|
}
|
|
)
|
|
|
|
@pytest.mark.slow
|
|
@pytest.mark.tts
|
|
@hardware_test(res={"cuda": "L4", "rocm": "MI325"}, num_cards=1)
|
|
def test_inline_pause_mid_text(self, omni_server, online_client) -> None:
|
|
"""Positional pause token placed inline between two clauses."""
|
|
online_client.send_audio_speech_request(
|
|
{
|
|
"model": omni_server.model,
|
|
"input": "Hold on a moment <|prosody:pause|> let me think about it.",
|
|
"stream": False,
|
|
"response_format": "wav",
|
|
"timeout": DEFAULT_SPEECH_TIMEOUT_S,
|
|
"min_audio_bytes": _MIN_AUDIO_BYTES,
|
|
"transcript_expected_text": "Hold on a moment. Let me think about it.",
|
|
}
|
|
)
|
|
|
|
@pytest.mark.slow
|
|
@pytest.mark.tts
|
|
@hardware_test(res={"cuda": "L4", "rocm": "MI325"}, num_cards=1)
|
|
def test_inline_sfx_with_onomatopoeia(self, omni_server, online_client) -> None:
|
|
"""SFX token paired with its written onomatopoeia (``<|sfx:laughter|>Hehe``)."""
|
|
online_client.send_audio_speech_request(
|
|
{
|
|
"model": omni_server.model,
|
|
"input": (
|
|
"<|emotion:amusement|>I cannot believe that just happened. "
|
|
"<|sfx:laughter|>Hehe, I am still recovering from it."
|
|
),
|
|
"stream": False,
|
|
"response_format": "wav",
|
|
"timeout": DEFAULT_SPEECH_TIMEOUT_S,
|
|
"min_audio_bytes": _MIN_AUDIO_BYTES,
|
|
"transcript_expected_text": "I cannot believe that just happened. I am still recovering from it.",
|
|
}
|
|
)
|
|
|
|
|
|
@pytest.mark.parametrize("omni_server", TEST_PARAMS, indirect=True)
|
|
class TestHiggsAudioV3OnlineVoiceClone:
|
|
"""Voice clone via the two payload shapes the model serves."""
|
|
|
|
@pytest.mark.slow
|
|
@pytest.mark.tts
|
|
@hardware_test(res={"cuda": "L4", "rocm": "MI325"}, num_cards=1)
|
|
def test_voice_clone_ref_audio_ref_text(self, omni_server, online_client) -> None:
|
|
"""Canonical vllm-omni voice clone via ``ref_audio`` + ``ref_text``."""
|
|
online_client.send_audio_speech_request(
|
|
{
|
|
"model": omni_server.model,
|
|
"input": "Hello world.",
|
|
"ref_audio": _REF_AUDIO_URL,
|
|
"ref_text": _REF_TEXT,
|
|
"stream": False,
|
|
"response_format": "wav",
|
|
"timeout": DEFAULT_SPEECH_TIMEOUT_S,
|
|
}
|
|
)
|
|
|
|
@pytest.mark.slow
|
|
@pytest.mark.tts
|
|
@hardware_test(res={"cuda": "L4", "rocm": "MI325"}, num_cards=1)
|
|
def test_voice_clone_references_alias(self, omni_server, online_client) -> None:
|
|
"""BosonAI cookbook payload: ``references=[{audio_path, text}]``.
|
|
|
|
``normalize_references_alias`` (``protocol/audio.py``) is supposed to
|
|
translate the cookbook field into ``ref_audio`` / ``ref_text``; without
|
|
it the request would silently fall through to zero-shot synthesis.
|
|
"""
|
|
online_client.send_audio_speech_request(
|
|
{
|
|
"model": omni_server.model,
|
|
"input": "Hello world.",
|
|
"references": [{"audio_path": _REF_AUDIO_URL, "text": _REF_TEXT}],
|
|
"stream": False,
|
|
"response_format": "wav",
|
|
"timeout": DEFAULT_SPEECH_TIMEOUT_S,
|
|
}
|
|
)
|