Files
vllm-omni/vllm_omni/model_executor/models/personaplex/duplex/config.py
T
2026-09-27 03:02:24 +00:00

25 lines
1.1 KiB
Python

# SPDX-License-Identifier: Apache-2.0
# SPDX-FileCopyrightText: Copyright contributors to the vLLM-Omni project
"""Constants of the PersonaPlex full-duplex integration.
PersonaPlex (``nvidia/personaplex-7b-v1``) is a Moshi finetune: a pure-lockstep
full-duplex speech-to-speech model running at the Mimi codec frame rate
(12.5 Hz / 80 ms). The values are fixed by the pretrained checkpoint; they
live here so the rest of the package never hard-codes them.
"""
from __future__ import annotations
# Mimi codec constants (loaders.py: SAMPLE_RATE / FRAME_RATE).
SAMPLE_RATE = 24000
FRAME_RATE = 12.5
FRAME_SIZE = int(SAMPLE_RATE / FRAME_RATE) # 1920 samples per 80 ms frame
CHUNK_PERIOD_MS = int(round(1000 / FRAME_RATE)) # 80
# Default assistant persona shipped with PersonaPlex (offline.py:335).
DEFAULT_PERSONA = "You are a wise and friendly teacher. Answer questions or provide advice in a clear and engaging way."
# Bundled voice prompt used when a session names none.
DEFAULT_VOICE = "NATF2.pt"
__all__ = ["CHUNK_PERIOD_MS", "DEFAULT_PERSONA", "DEFAULT_VOICE", "FRAME_RATE", "FRAME_SIZE", "SAMPLE_RATE"]