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

whisper nits (#1388)

Browse Source
This commit is contained in:
Awni Hannun
2025-09-03 13:18:50 -07:00
committed by GitHub
parent c1af8c46bd
commit 8e4391ca21
2 changed files with 1 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
whisper/benchmark.py
View File

@@ -11,12 +11,6 @@ audio_file = "mlx_whisper/assets/ls_test.flac"
def parse_arguments(): def parse_arguments():
parser = argparse.ArgumentParser(description="Benchmark script.") parser = argparse.ArgumentParser(description="Benchmark script.")
parser.add_argument(
"--mlx-dir",
type=str,
default="mlx_models",
help="The folder of MLX models",
)
parser.add_argument( parser.add_argument(
"--all", "--all",
action="store_true", action="store_true",

2
whisper/mlx_whisper/transcribe.py
View File

@@ -62,7 +62,7 @@ class ModelHolder:
def transcribe( def transcribe(
audio: Union[str, np.ndarray, mx.array], audio: Union[str, np.ndarray, mx.array],
*, *,
path_or_hf_repo: str = "mlx-community/whisper-tiny", path_or_hf_repo: str = "mlx-community/whisper-turbo",
verbose: Optional[bool] = None, verbose: Optional[bool] = None,
temperature: Union[float, Tuple[float, ...]] = (0.0, 0.2, 0.4, 0.6, 0.8, 1.0), temperature: Union[float, Tuple[float, ...]] = (0.0, 0.2, 0.4, 0.6, 0.8, 1.0),
compression_ratio_threshold: Optional[float] = 2.4, compression_ratio_threshold: Optional[float] = 2.4,