diff --git a/src/runners/audio_transcriber.py b/src/runners/audio_transcriber.py index 87c5456..4140904 100644 --- a/src/runners/audio_transcriber.py +++ b/src/runners/audio_transcriber.py @@ -44,7 +44,7 @@ def __init__(self, model_id: str = "openai/whisper-large-v3"): def forward(self, data: List[bytes]) -> List[str]: start = time.time() - result = self.pipe(data) + result = self.pipe(data, batch_size=len(data)) transcribe_time = time.time() - start return [ BatchOutput(