https://github.com/runpod-workers/worker-whisper
🎙️ | Whisper Worker for RunPod: Processing and transcribing audio with various Whisper models. Part of RunPod Workers collection.
https://github.com/runpod-workers/worker-whisper
audio-processing language-translation runpod runpod-worker transcription whisper
Last synced: 12 months ago
JSON representation
🎙️ | Whisper Worker for RunPod: Processing and transcribing audio with various Whisper models. Part of RunPod Workers collection.
- Host: GitHub
- URL: https://github.com/runpod-workers/worker-whisper
- Owner: runpod-workers
- License: mit
- Created: 2023-05-06T00:58:58.000Z (about 3 years ago)
- Default Branch: main
- Last Pushed: 2024-10-24T20:36:46.000Z (over 1 year ago)
- Last Synced: 2025-05-30T09:27:51.535Z (about 1 year ago)
- Topics: audio-processing, language-translation, runpod, runpod-worker, transcription, whisper
- Language: Python
- Homepage: https://docs.runpod.io/reference/whisper
- Size: 16.6 KB
- Stars: 6
- Watchers: 3
- Forks: 10
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
Whisper | Worker
This repository contains the Whisper Worker for RunPod. The Whisper Worker is designed to process audio files using various Whisper models, with options for transcription formatting, language translation, and more. It's part of the RunPod Workers collection aimed at providing diverse functionality for endpoint processing.
[](https://github.com/runpod-workers/worker-template/actions/workflows/CD-docker_dev.yml)
## Model Inputs
| Input | Type | Description |
|-------------------------------------|-------|-------------------------------------------------------------------------------------------------------------|
| `audio` | Path | Audio file |
| `model` | str | Choose a Whisper model. Choices: "tiny", "base", "small", "medium", "large-v1", "large-v2". Default: "base" |
| `transcription` | str | Choose the format for the transcription. Choices: "plain text", "srt", "vtt". Default: "plain text" |
| `translate` | bool | Translate the text to English when set to True. Default: False |
| `language` | str | Language spoken in the audio, specify None to perform language detection. Default: None |
| `temperature` | float | Temperature to use for sampling. Default: 0 |
| `best_of` | int | Number of candidates when sampling with non-zero temperature. Default: 5 |
| `beam_size` | int | Number of beams in beam search, only applicable when temperature is zero. Default: 5 |
| `patience` | float | Optional patience value to use in beam decoding. Default: None |
| `length_penalty` | float | Optional token length penalty coefficient (alpha). Default: None |
| `suppress_tokens` | str | Comma-separated list of token ids to suppress during sampling. Default: "-1" |
| `initial_prompt` | str | Optional text to provide as a prompt for the first window. Default: None |
| `condition_on_previous_text` | bool | If True, provide the previous output of the model as a prompt for the next window. Default: True |
| `temperature_increment_on_fallback` | float | Temperature to increase when falling back when the decoding fails. Default: 0.2 |
| `compression_ratio_threshold` | float | If the gzip compression ratio is higher than this value, treat the decoding as failed. Default: 2.4 |
| `logprob_threshold` | float | If the average log probability is lower than this value, treat the decoding as failed. Default: -1.0 |
| `no_speech_threshold` | float | If the probability of the token is higher than this value, consider the segment as silence. Default: 0.6 |
## Test Inputs
The following inputs can be used for testing the model:
```json
{
"input": {
"audio": "https://github.com/runpod-workers/sample-inputs/raw/main/audio/gettysburg.wav"
}
}
```