You can deploy this model and start earning money today!

kerasformers/
whisper_tiny

Name: whisper_tiny
Rating: 2.5 (68 reviews)

kerasformers

keras

whisper

automatic-speech-recognition

Views

No views yet

See our collection for all versions of Whisper.

Run Whisper with Keras 3: JAX, PyTorch, or TensorFlow

GitHub

zeromodels/whisper_tiny

Paper: Robust Speech Recognition via Large-Scale Weak Supervision (arXiv:2212.04356) · HF Papers

Whisper is a multilingual encoder-decoder ASR model trained on large-scale weak supervision. Use task="transcribe" to keep the source language or task="translate" to render English. Pass language=None to let the model detect the spoken language. Output is cased and punctuated.

For more details on the model, please go to the upstream model card.

Pure-Keras 3 conversion of openai/whisper-tiny for zeromodels. One implementation runs unmodified on TensorFlow / Torch / JAX.

This is an ASR checkpoint (WhisperConditionalGenerate, 39M).

✨ Quick start

python

import os
os.environ["KERAS_BACKEND"] = "torch"  # or "jax" / "tensorflow"

import soundfile as sf
from zeromodels.models.whisper import (
    WhisperProcessor,
    WhisperConditionalGenerate,
)

model = WhisperConditionalGenerate.from_weights("zeromodels/whisper_tiny")
processor = WhisperProcessor.from_weights("zeromodels/whisper_tiny")

audio, sr = sf.read("your_audio.wav", dtype="float32")  # 16 kHz mono
# task="transcribe" keeps the source language; "translate" -> English.
text = model.generate(audio, processor, language="en", task="transcribe")
print(repr(text[0]))

Load any Whisper variant the same way with from_weights("zeromodels/<variant>"):

Variant	Hub	Notes
`whisper_tiny`	`zeromodels/whisper_tiny`	39M
`whisper_base`	`zeromodels/whisper_base`	74M
`whisper_small`	`zeromodels/whisper_small`	244M
`whisper_medium`	`zeromodels/whisper_medium`	769M
`whisper_large`	`zeromodels/whisper_large`	1.55B
`whisper_large_v2`	`zeromodels/whisper_large_v2`	1.55B
`whisper_large_v3`	`zeromodels/whisper_large_v3`	128 mel bins
`whisper_large_v3_turbo`	`zeromodels/whisper_large_v3_turbo`	4 decoder layers

Tips

Set KERAS_BACKEND before importing Keras / zeromodels.
Prefer WhisperProcessor.from_weights(...) so mel bins match the variant (v3 uses 128).
Clips are padded to a 30 s window; chunk longer audio yourself.
See Whisper docs and Loading Weights.
Community / upstream safetensors still work via the hf: prefix, e.g. WhisperConditionalGenerate.from_weights("hf:openai/whisper-tiny").

Special Thanks

A huge thank you to the OpenAI Whisper authors for creating and releasing these models.

License: Apache 2.0.

kerasformers/whisper_tiny