TheStageAI/TheWhisper

Optimized Whisper models for streaming and on-device use

What it solves

TheWhisper provides a high-performance, low-latency speech-to-text (STT) solution optimized for self-hosting and on-device inference. It addresses the limitation of original Whisper models that require 30-second audio chunks, allowing for flexible chunk sizes (10s, 15s, 20s, and 30s) without needing to pad audio with silence.

How it works

The project provides fine-tuned versions of Whisper models with open weights and specialized inference engines for different hardware:

  • NVIDIA GPUs: Uses optimized engines (including TheStage AI ElasticModels) to achieve high throughput, such as 220 tok/s on L40s GPUs.
  • Apple Silicon: Uses CoreML and MLX engines to minimize power consumption (2W) and RAM usage (2GB).
  • Streaming: Implements streaming transcription for both NVIDIA and macOS platforms to support real-time applications.

Who it’s for

It is designed for developers building real-time captioning, live meeting tools, voice interfaces, and edge deployments where low power and low latency are are required.

Highlights

  • Flexible Chunking: Supports 10s, 15s, 20s, and 30s audio chunks for more efficient processing.
  • Hardware Optimization: Dedicated high-performance engines for NVIDIA GPUs (RTX 4090/5090, L40s, H100, A100) and Apple Silicon (M1-M4 series).
  • On-Device Deployment: Includes a Python API and examples for building macOS desktop applications using Electron and ReactJS.
  • Streaming Support: Built-in support for streaming transcription across supported platforms.

Related

  • Project
  • Project
  • Project
  • Project
  • Project