zhangyiss/mlx-examples
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-09-01 12:49:50 +08:00
Code Issues 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 ab576be362
commit 41ab0cd055
1 changed files with 3 additions and 3 deletions
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) return decoding.decode(model, mels)
def everything(): def everything(model_name):
return transcribe(audio_file) return transcribe(audio_file, model=model_name)
if __name__ == "__main__": if __name__ == "__main__":
@@ -99,6 +99,6 @@ if __name__ == "__main__":
print(f"Model forward time {model_forward_time:.3f}") print(f"Model forward time {model_forward_time:.3f}")
decode_time = timer(decode, model, mels) decode_time = timer(decode, model, mels)
print(f"Decode time {decode_time:.3f}") 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"Everything time {everything_time:.3f}")
print(f"\n{'-----' * 10}\n") print(f"\n{'-----' * 10}\n")