Dmitriy Dementiev
0af2dbdd17
- Зачем: - добавлена локальная транскрипция смешанной речи и русского текста с пунктуацией. - Что: - onnx-asr обновлён до 0.12 и зарегистрированы три модели GigaAM. - выбор compute_type учитывает опубликованные квантизации и явность настройки. - обновлены тесты, README и требования PRD. - Проверка: - uv run pytest -q: 223 passed, 1 skipped. - выполнены smoke- и полные прогоны четырёх GigaAM-моделей.
41 lines
1.6 KiB
Python
41 lines
1.6 KiB
Python
"""Реестр бэкендов транскрипции и выбор бэкенда по устройству."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import TYPE_CHECKING
|
|
|
|
if TYPE_CHECKING:
|
|
from .base import Backend
|
|
|
|
|
|
def get_backend(device: str, *, compute_type_explicit: bool = True) -> Backend:
|
|
"""Возвращает экземпляр бэкенда для указанного устройства.
|
|
|
|
Импорты ленивые — бэкенд загружается только при запросе.
|
|
compute_type_explicit: False если compute_type пришёл из дефолтов (влияет на fallback).
|
|
"""
|
|
if device in ("openvino", "openvino-gpu", "openvino-cpu"):
|
|
try:
|
|
from .openvino import OpenVINOBackend
|
|
except ImportError:
|
|
raise ValueError(
|
|
"OpenVINO бэкенд недоступен. Установите: pip install openvino-genai"
|
|
) from None
|
|
return OpenVINOBackend(
|
|
ov_device=device, compute_type_explicit=compute_type_explicit
|
|
)
|
|
|
|
if device == "onnx":
|
|
try:
|
|
from .onnx_asr import OnnxAsrBackend
|
|
except ImportError:
|
|
raise ValueError(
|
|
"onnx-asr бэкенд недоступен. Установите: pip install onnx-asr[cpu,hub]"
|
|
) from None
|
|
return OnnxAsrBackend(compute_type_explicit=compute_type_explicit)
|
|
|
|
# cuda, cpu и всё остальное → faster-whisper
|
|
from .faster_whisper import FasterWhisperBackend
|
|
|
|
return FasterWhisperBackend()
|