kairos-asr 0.6.3

High-performance Russian speech recognition library based on GigaAM-style RNN-T and ONNX.

Kairos Automatic Speech Recognition

---

Kairos ASR — высокопроизводительная библиотека распознавания русской речи на базе GigaAM-style RNN-T архитектуры и ONNX.

Проект сфокусирован на скорости, точности и удобстве интеграции в микросервисы и десктопные приложения.

📄 Описание

---

• Оптимизированный ONNX-инференс
• Работает на CPU и GPU (CUDA, extra [gpu])
• Поддержка временных меток (word-level, sentence-level)
• Итеративная обработка с выводом прогресса и ETA
• Встроенный Voice-Activity-Detection (Silero VAD)
• Поддержка длинных аудио
• Простая установка и использование
• Поддержка Windows и Linux

⚡ TL;DR

• pip install kairos-asr
• Запустить: kairos-asr transcribe example.wav или см. Python сниппет ниже.
• Полное руководство: docs/USAGE.md.

📦 Установка

Веса доступны на Hugging Face: Alenkar/KairosASR

🖥️ Системные требования

• ffmpeg должен быть доступен в PATH (используется для загрузки и ресемплинга аудио).
• Доступ в интернет. При первом запуске скачиваются веса моделей с Hugging Face и модель Silero VAD.
• Для ускорения скачиваний и избежания лимитов рекомендуется установить токен HF: export HF_TOKEN=... (или huggingface-cli login).

Быстрый старт (CPU)

pip install kairos-asr

Другие способы установки

Через GIT

pip install git+https://github.com/Alenkar/kairos-asr.git
# или
git clone https://github.com/Alenkar/kairos-asr.git
cd kairos-asr
pip install -e .

Поддержка GPU (CUDA)

1. Пакет с GPU-опциями:

pip install "kairos-asr[gpu]"

2. Torch/Torchaudio под вашу версию CUDA:

# пример под CUDA 12.1/12.2 (cu121)
pip install torch==2.6.0 torchaudio==2.6.0 \
  --index-url https://download.pytorch.org/whl/cu121 --upgrade

Найдите свой индекс (cu118, cu121 и т.д.) на pytorch.org и подставьте в команду.

🚀 Использование (Python)

Минимальный пример:

from kairos_asr import KairosASR

asr = KairosASR()  # авто-загрузка весов
result = asr.transcribe(wav_file="audio.wav")
print(result.full_text)

Требования к входному аудио:

• Файлы, поддерживаемые ffmpeg; автоматический ресемплинг до 16 kHz.
• Рекомендуется WAV PCM 16-bit, mono, 16 kHz; стерео приводится к моно.
• Длинные записи режутся Silero VAD на ~15–25 c (жёсткий лимит ~30 c) и объединяются.

Использование (CLI)

Существует несколько команд для работы через терминал.

# Показать информацию
kairos-asr info

# Проверить окружение и вывести информацию
kairos-asr doctor

# Список моделей (показывает локальное наличие и путь)
kairos-asr list

# Скачать все модели
kairos-asr download

# Скачать только encoder
kairos-asr download encoder

# Перевести файл в текст
kairos-asr transcribe <wav_file>

🧠 Как работает

---

1. VAD делит аудио на сегменты (до 25 секунд).
2. Каждый сегмент преобразуется в Mel-spectrogram.
3. RNN-T модель (Encoder + Predictor + Joint).
4. Выравнивание слов по фреймам.
5. Объединение сегментов в финальную разметку.

Архитектура и общий пайплайн вдохновлены проектом GigaAM.

📜 Лицензия и происхождение

---

Код

Часть исходного кода основана на проекте GigaAM и используется в соответствии с условиями лицензии MIT. Оригинальный проект GigaAM:

• License: MIT
• Copyright (c) 2024 GigaChat Team

Модели

Используемые веса моделей происходят из проекта GigaAM, но были:

• конвертированы в формат ONNX,
• оптимизированы для CPU/GPU-inference,
• адаптированы для использования в Kairos ASR,
• частично произведена тонкая настройка (Lora) на кастомных данных.

Права на исходные веса принадлежат правообладателям проекта GigaAM. Использование осуществляется в рамках лицензии MIT.

Если вы используете KairosASR в исследовательских или коммерческих целях — проверьте совместимость с лицензией оригинальной модели (GigaAM).

📊 Планы развития

---

• Batch-inference.
• Realtime-inference (микрофон) и чанки с ndarray.
• Diarization (разделение спикеров).
• Тесты.
• Дальнейшее дообучение на специализированных данных.
• Скрипт для Lora finetune на своих данных.
• Увеличение Vocabulary.
• Пакет для установки.
• Веса на Hugging Face.

👤 Автор

---

Разработано и поддерживается: (Alexey Shimokhin / Alenkar).
Если вы используете Kairos ASR в своих проектах — ссылка или отзыв приветствуются.