Aller au contenu

asr_transformers model guide

Overview

asr_transformers is a VoiceHub automatic speech recognition integration. This page is generated from the model registry and its executable data and training contracts, so the documented support stays aligned with code. Open the asr_transformers Colab notebook.

Quickstart

python -m pip install voicehub
  1. Install VoiceHub and the provider extra shown above.
  2. Choose a checkpoint that matches this integration.
  3. Place a supported recording at speech.wav.
  4. Transcribe it and inspect both the full text and timed segments.
from voicehub import AutoModelForSpeechRecognition

model = AutoModelForSpeechRecognition.from_pretrained(
    'openai/whisper-small',
    model_type='asr_transformers',
    device="cuda",
    lazy_load=True,
)
output = model.transcribe("speech.wav")
print(output.text)
for segment in output.segments:
    print(segment.start, segment.end, segment.text)

Use only authorized recordings for reference voice, transcription, detection, or evaluation. The example selects a concrete device; verify checkpoint-specific hardware needs and pin an immutable revision before production use.

Supported tasks and capabilities

Property Value
Task Automatic speech recognition
Architecture native-asr-dispatch
Runtime VoiceHub-native
Capabilities automatic-speech-recognition, multilingual, timestamps, safetensors, fine-tuning, ctc, speech-seq2seq, voicehub-native, native-runtime
Reusable components

Data contract

Property Value
Readiness integrated-raw
Data architecture native-dispatch
Sample rate 16,000 Hz
Contract getter get_asr_dataset_spec('asr_transformers')
Variant Required fields One of Boundary Other rules
raw-audio audio text / transcription / transcript Source at most one: text / transcription / transcript
feature-model-ready input_features, labels Prepared
waveform-model-ready input_values, labels Prepared

Checkpoint-dispatched raw and cached inputs for native Transformers ASR families. Follow the shared data workflow for manifest loading, audio validation, leakage-safe splits, and model-owned preprocessing.

Checkpoints, provenance, and license

Property Value
Default checkpoint openai/whisper-small
Checkpoint status Registry default; pin an immutable revision for production and reproducible evidence
Implementation voicehub.models.asr_transformers.modeling_asr_transformers.TransformersASRForSpeechRecognition
Configuration voicehub.models.asr_transformers.configuration_asr_transformers.TransformersASRConfig
Source provenance voicehub/architectures/moonshine/SOURCE.json
License Checkpoint-specific

No VoiceHub-specific license override is registered. Verify the checkpoint and upstream source terms before use.

The default checkpoint identifies the expected family, not every compatible variant. Confirm the selected checkpoint's revision, access terms, provenance, and license before downloading or redistributing it.

Optimization and training support

All public optimizations enter this model through the shared BaseSpeechModel lifecycle. Use available_optimization_passes() to discover the public pass registry, then apply, inspect, serialize, or restore a plan through the common model API. Application remains fail-closed when the active runtime or hardware cannot satisfy a pass.

Training contract

Property Value
Support native
Family native-asr-dispatch
Recipe single-phase
Default phase speech_recognition
Training checkpoint openai/whisper-small
Native training graph yes
Phase Kind Components Required inputs Loss keys
speech_recognition objective model loss

The integration accepts its declared source or prepared contract directly. Call model.validate_training_support() before constructing a trainer. Follow the shared training workflow for a one-step smoke test, validation, checkpoint resume, optimization, and portable export.

Public API

Purpose Public object
Discover get_model_spec('asr_transformers')
Load and run AutoModelForSpeechRecognition
Configure TransformersASRConfig
Model implementation TransformersASRForSpeechRecognition
Normalized output ASROutput
Training contract get_training_spec('asr_transformers')
Optimization lifecycle available_optimization_passes, apply_optimization_plan, optimization_manifest, restore_optimization_plan

Related shared documentation: