Skip to main content

CLI for Omi Med STT v1 medical speech-to-text

Project description

Omi Med STT Runtime

Runtime CLI for Omi Med STT v1, an English medical speech-to-text model.

This repository contains runtime code only. It does not contain model weights, private benchmark data, or training data.

Runtimes

omi-med-stt supports three runtime paths:

Runtime Best for Artifact
mlx Apple Silicon Macs omi-health/omi-med-stt-v1-mlx
cpp Windows, Mac Intel, Linux CPU, and ggml GPU backends omi-health/omi-med-stt-v1-gguf
nemo NVIDIA CUDA servers and canonical NeMo checkpoint use omi-health/omi-med-stt-v1

The source-of-truth model is the NeMo checkpoint. MLX and GGUF are runtime exports.

Install

From PyPI:

pip install -U omi-med-stt

From this repository:

pip install git+https://github.com/Omi-Health/omi-med-stt-runtime.git

For Apple Silicon / MLX:

pip install "omi-med-stt[mlx] @ git+https://github.com/Omi-Health/omi-med-stt-runtime.git"

For CUDA/Linux NeMo:

pip install "omi-med-stt[nemo] @ git+https://github.com/Omi-Health/omi-med-stt-runtime.git"

Basic Usage

Simple path:

omi-med-stt audio.wav

Explicit MLX:

omi-med-stt audio.wav --runtime mlx

Explicit NeMo:

omi-med-stt audio.wav --runtime nemo

Explicit parakeet.cpp / GGUF:

omi-med-stt audio.wav --runtime cpp

JSON output:

omi-med-stt audio.wav --json

Dependency/runtime check:

omi-med-stt check

parakeet.cpp / GGUF Runtime

The cpp runtime is powered by parakeet.cpp, a C++/ggml inference engine for NVIDIA Parakeet ASR models.

Omi Med STT v1 includes a post-Conformer medical adapter. Until this adapter extension is upstreamed, omi-med-stt builds parakeet.cpp with the adapter patch included in this repository and caches the resulting parakeet-cli.

Normal use:

omi-med-stt audio.wav --runtime cpp

Pre-build the C++ runtime explicitly:

omi-med-stt install-cpp

Choose a backend:

omi-med-stt install-cpp --cpp-backend cpu
omi-med-stt install-cpp --cpp-backend metal
omi-med-stt install-cpp --cpp-backend cuda

Manual override remains available for developers:

omi-med-stt audio.wav --runtime cpp --parakeet-cli /path/to/parakeet-cli

The cached runtime is built from parakeet.cpp and applies parakeet-cpp-omi-adapter.patch. You need git and cmake available for the first build.

The cpp runtime downloads only the selected GGUF file. It does not download the NeMo .nemo checkpoint or the MLX model.safetensors.

By default, the cpp runtime is pinned to a specific Hugging Face repository revision and verifies the SHA256 checksum for the official f16/q8_0 GGUF files. You can override this for experiments:

omi-med-stt audio.wav --runtime cpp --revision main
omi-med-stt audio.wav --runtime cpp --no-verify-checksum

Long Audio

Omi Med STT v1 is based on Parakeet and can handle long-form audio. Start with the simple path:

omi-med-stt consult.wav

An explicit chunked path is still available for constrained environments:

omi-med-stt transcribe-long consult.wav --chunk-seconds 25 --overlap 3

Model Access

The runtime defaults to these Hugging Face model repositories:

  • omi-health/omi-med-stt-v1
  • omi-health/omi-med-stt-v1-mlx
  • omi-health/omi-med-stt-v1-gguf

If the model repositories are private before launch, authenticate first:

huggingface-cli login

Attribution

This runtime uses or interoperates with:

  • NVIDIA NeMo / Parakeet, for the base ASR architecture.
  • parakeet-mlx, for Apple Silicon MLX inference.
  • parakeet.cpp, for GGUF / C++ / ggml inference.

See NOTICE.md.

License

Runtime code in this repository is MIT licensed.

Model weights are governed separately by the model repositories. Omi Med STT v1 is a derivative of nvidia/parakeet-tdt-0.6b-v2, whose model weights are licensed under CC-BY-4.0.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

omi_med_stt-0.1.8.tar.gz (18.4 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

omi_med_stt-0.1.8-py3-none-any.whl (18.2 kB view details)

Uploaded Python 3

File details

Details for the file omi_med_stt-0.1.8.tar.gz.

File metadata

  • Download URL: omi_med_stt-0.1.8.tar.gz
  • Upload date:
  • Size: 18.4 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.12.9

File hashes

Hashes for omi_med_stt-0.1.8.tar.gz
Algorithm Hash digest
SHA256 1e89086fb530c2f0ce8e2a008004bcac5e688984b36cdc0c09492b62980e868b
MD5 a51a68627cb661d7c22ec184f6bb0193
BLAKE2b-256 781c0a819d90ecfed9a1de76b52662d2b96b74f3d78f07af68a64172a31e634d

See more details on using hashes here.

File details

Details for the file omi_med_stt-0.1.8-py3-none-any.whl.

File metadata

  • Download URL: omi_med_stt-0.1.8-py3-none-any.whl
  • Upload date:
  • Size: 18.2 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.12.9

File hashes

Hashes for omi_med_stt-0.1.8-py3-none-any.whl
Algorithm Hash digest
SHA256 5a44d165e4e1c0b49dd454a43a7beb768891d35a63e492b548af43055eca43f6
MD5 1b17fbb556fabdfd2afa88ff7243947e
BLAKE2b-256 cfa5246aade955a6e5694ca603987fe228e97989be841f40cc13e13329cd8dd6

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page