30 lines
844 B
Python
30 lines
844 B
Python
"""Custom vLLM config parsers for vllm-omni."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import importlib
|
|
|
|
_CLASS_TO_MODULE: dict[str, str] = {
|
|
"VoxtralTTSConfigParser": "vllm_omni.transformers_utils.parsers.voxtral_tts",
|
|
}
|
|
|
|
__all__ = ["VoxtralTTSConfigParser"]
|
|
|
|
|
|
def __getattr__(name: str):
|
|
if name in _CLASS_TO_MODULE:
|
|
module_name = _CLASS_TO_MODULE[name]
|
|
module = importlib.import_module(module_name)
|
|
return getattr(module, name)
|
|
|
|
raise AttributeError(f"module 'vllm_omni.transformers_utils.parsers' has no attribute {name!r}")
|
|
|
|
|
|
def __dir__():
|
|
return sorted(list(__all__))
|
|
|
|
|
|
# Eagerly import parser modules so their registry side-effects run as soon as
|
|
# `vllm_omni.transformers_utils.parsers` is imported.
|
|
from vllm_omni.transformers_utils.parsers import voxtral_tts as _voxtral_tts # noqa: F401, E402
|