liuhuayong/mlx-examples
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix benchmark (#200)

Browse Source
This commit is contained in:
bofeng huang 2023-12-28 20:29:39 +01:00 committed by GitHub
parent 78d207fe27
commit e1e56a625b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
whisper/benchmark.py
View File

@ -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")