support batch processing

This commit is contained in:
Tengda Han
2023-02-01 19:41:20 +00:00
parent fd2a093754
commit 039af89a86
3 changed files with 234 additions and 3 deletions

View File

@ -11,7 +11,7 @@ from tqdm import tqdm
from .audio import load_audio, log_mel_spectrogram, pad_or_trim
from .decoding import DecodingOptions, DecodingResult, decode, detect_language
from .model import Whisper, ModelDimensions
from .transcribe import transcribe, transcribe_with_vad
from .transcribe import transcribe, transcribe_with_vad, transcribe_with_vad_parallel
from .alignment import load_align_model, align
_MODELS = {
"tiny.en": "https://openaipublic.azureedge.net/main/whisper/models/d3dd57d32accea0b295c96e26691aa14d8822fac7d9d27d5dc00b4ca2826dd03/tiny.en.pt",