wx44wx/pipecat
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

WhisperSTTService: Add additional whisper model variants

Browse Source
This commit is contained in:
shahrukhx01
2025-07-07 07:40:33 +02:00
committed by shahrukh.khan
parent 6eed6ff779
commit a84e7e30da
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/pipecat/services/whisper/stt.py
View File

@@ -49,8 +49,10 @@ class Model(Enum):
Parameters: Parameters:
TINY: Smallest multilingual model, fastest inference. TINY: Smallest multilingual model, fastest inference.
BASE: Basic multilingual model, good speed/quality balance. BASE: Basic multilingual model, good speed/quality balance.
SMALL: Small multilingual model, better speed/quality balance than BASE.
MEDIUM: Medium-sized multilingual model, better quality. MEDIUM: Medium-sized multilingual model, better quality.
LARGE: Best quality multilingual model, slower inference. LARGE: Best quality multilingual model, slower inference.
LARGE_V3_TURBO: Fast multilingual model, slightly lower quality than LARGE.
DISTIL_LARGE_V2: Fast multilingual distilled model. DISTIL_LARGE_V2: Fast multilingual distilled model.
DISTIL_MEDIUM_EN: Fast English-only distilled model. DISTIL_MEDIUM_EN: Fast English-only distilled model.
""" """
@@ -58,8 +60,10 @@ class Model(Enum):
# Multilingual models # Multilingual models
TINY = "tiny" TINY = "tiny"
BASE = "base" BASE = "base"
SMALL = "small"
MEDIUM = "medium" MEDIUM = "medium"
LARGE = "large-v3" LARGE = "large-v3"
LARGE_V3_TURBO = "deepdml/faster-whisper-large-v3-turbo-ct2"
DISTIL_LARGE_V2 = "Systran/faster-distil-whisper-large-v2" DISTIL_LARGE_V2 = "Systran/faster-distil-whisper-large-v2"
# English-only models # English-only models