From 8e4391ca215e07f80e2ead1f3487b17b54906007 Mon Sep 17 00:00:00 2001 From: Awni Hannun Date: Wed, 3 Sep 2025 13:18:50 -0700 Subject: [PATCH] whisper nits (#1388) --- whisper/benchmark.py | 6 ------ whisper/mlx_whisper/transcribe.py | 2 +- 2 files changed, 1 insertion(+), 7 deletions(-) diff --git a/whisper/benchmark.py b/whisper/benchmark.py index 8101e343..3f6a092e 100644 --- a/whisper/benchmark.py +++ b/whisper/benchmark.py @@ -11,12 +11,6 @@ audio_file = "mlx_whisper/assets/ls_test.flac" def parse_arguments(): 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( "--all", action="store_true", diff --git a/whisper/mlx_whisper/transcribe.py b/whisper/mlx_whisper/transcribe.py index e9c2751f..bced16a5 100644 --- a/whisper/mlx_whisper/transcribe.py +++ b/whisper/mlx_whisper/transcribe.py @@ -62,7 +62,7 @@ class ModelHolder: def transcribe( 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, temperature: Union[float, Tuple[float, ...]] = (0.0, 0.2, 0.4, 0.6, 0.8, 1.0), compression_ratio_threshold: Optional[float] = 2.4,