vllm-omni/vllm_omni/transformers_utils/configs/voxtral_tts.py

from __future__ import annotations

from typing import Any

from transformers import AutoConfig, PretrainedConfig


class VoxtralTTSConfig(PretrainedConfig):
    """HuggingFace-style config for Voxtral TTS models."""

    model_type = "voxtral_tts"

    def __init__(
        self,
        text_config: PretrainedConfig | dict | None = None,
        audio_config: dict[str, Any] | None = None,
        **kwargs: Any,
    ) -> None:
        # Assign before super().__init__: transformers/huggingface_hub validation
        # (e.g. validate_token_ids → get_text_config) runs during base __init__.
        if isinstance(text_config, PretrainedConfig):
            self.text_config = text_config
        elif isinstance(text_config, dict):
            self.text_config = PretrainedConfig.from_dict(text_config)
        else:
            self.text_config = PretrainedConfig()

        self.audio_config = audio_config or {}
        super().__init__(**kwargs)

    def get_text_config(self, **kwargs: Any) -> PretrainedConfig:
        try:
            return object.__getattribute__(self, "text_config")
        except AttributeError:
            tc = PretrainedConfig()
            object.__setattr__(self, "text_config", tc)
            return tc


AutoConfig.register("voxtral_tts", VoxtralTTSConfig)

__all__ = ["VoxtralTTSConfig"]