From e1e56a625b801109e82d413cfa6e7e6db0f6a250 Mon Sep 17 00:00:00 2001 From: bofeng huang Date: Thu, 28 Dec 2023 20:29:39 +0100 Subject: [PATCH] Fix benchmark (#200) --- whisper/benchmark.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/whisper/benchmark.py b/whisper/benchmark.py index 10025952..877bb4f0 100644 --- a/whisper/benchmark.py +++ b/whisper/benchmark.py @@ -41,8 +41,8 @@ def decode(model, mels): return decoding.decode(model, mels) -def everything(): - return transcribe(audio_file) +def everything(model_name): + return transcribe(audio_file, model=model_name) if __name__ == "__main__": @@ -99,6 +99,6 @@ if __name__ == "__main__": print(f"Model forward time {model_forward_time:.3f}") decode_time = timer(decode, model, mels) print(f"Decode time {decode_time:.3f}") - everything_time = timer(everything) + everything_time = timer(everything, model_name) print(f"Everything time {everything_time:.3f}") print(f"\n{'-----' * 10}\n")