Skip to content

Commit 8e4391c

Browse files
authored
whisper nits (#1388)
1 parent c1af8c4 commit 8e4391c

File tree

2 files changed

+1
-7
lines changed

2 files changed

+1
-7
lines changed

whisper/benchmark.py

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -11,12 +11,6 @@
1111

1212
def parse_arguments():
1313
parser = argparse.ArgumentParser(description="Benchmark script.")
14-
parser.add_argument(
15-
"--mlx-dir",
16-
type=str,
17-
default="mlx_models",
18-
help="The folder of MLX models",
19-
)
2014
parser.add_argument(
2115
"--all",
2216
action="store_true",

whisper/mlx_whisper/transcribe.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,7 @@ def get_model(cls, model_path: str, dtype: mx.Dtype):
6262
def transcribe(
6363
audio: Union[str, np.ndarray, mx.array],
6464
*,
65-
path_or_hf_repo: str = "mlx-community/whisper-tiny",
65+
path_or_hf_repo: str = "mlx-community/whisper-turbo",
6666
verbose: Optional[bool] = None,
6767
temperature: Union[float, Tuple[float, ...]] = (0.0, 0.2, 0.4, 0.6, 0.8, 1.0),
6868
compression_ratio_threshold: Optional[float] = 2.4,

0 commit comments

Comments
 (0)