Skip to content

Commit

Permalink
add batch_size parameter
Browse files Browse the repository at this point in the history
  • Loading branch information
evq committed Sep 10, 2024
1 parent 649f12d commit 9f3e14f
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/runners/audio_transcriber.py
Original file line number Diff line number Diff line change
Expand Up @@ -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(
Expand Down

0 comments on commit 9f3e14f

Please sign in to comment.