zhangyiss/mlx-examples
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-10-23 22:18:06 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Whisper: Add pip distribution configuration to support pip installations. (#739)

Browse Source 
* Whisper: rename whisper to mlx_whisper

* Whisper: add setup.py config for publish

* Whisper: add assets data to setup config

* Whisper: pre-commit for setup.py

* Whisper: Update README.md

* Whisper: Update README.md

* nits

* fix package data

* nit in readme

---------

Co-authored-by: Awni Hannun <awni@apple.com>
This commit is contained in:
madroid
2024-05-02 00:00:02 +08:00
committed by GitHub
parent 4bf2eb17f2
commit 6775d6cb3f
23 changed files with 116 additions and 74 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
whisper/benchmark.py
View File

@@ -1,14 +1,12 @@
# Copyright © 2023 Apple Inc.
# Copyright © 2023-2024 Apple Inc.
import argparse
import os
import subprocess
import time
import mlx.core as mx
from mlx_whisper import audio, decoding, load_models, transcribe
from whisper import audio, decoding, load_models, transcribe
audio_file = "whisper/assets/ls_test.flac"
audio_file = "mlx_whisper/assets/ls_test.flac"
def parse_arguments():
@@ -83,16 +81,7 @@ if __name__ == "__main__":
print(f"\nFeature time {feat_time:.3f}")
for model_name in models:
model_path = f"{args.mlx_dir}/{model_name}"
if not os.path.exists(model_path):
print(
f"\nDidn't find the MLX-format {model_name} model in the folder {args.mlx_dir}. Lauching conversion"
)
subprocess.run(
f"python convert.py --torch-name-or-path {model_name} --mlx-path {model_path}",
shell=True,
)
model_path = f"mlx-community/whisper-{model_name}-mlx"
print(f"\nModel: {model_name.upper()}")
tokens = mx.array(
[