diff --git a/faster_whisper/transcribe.py b/faster_whisper/transcribe.py index d270bdf4..f09b60c2 100644 --- a/faster_whisper/transcribe.py +++ b/faster_whisper/transcribe.py @@ -594,6 +594,7 @@ def __init__( cpu_threads: int = 0, num_workers: int = 1, download_root: Optional[str] = None, + output_dir: Optional[str] = None, local_files_only: bool = False, files: dict = None, **model_kwargs, @@ -640,6 +641,7 @@ def __init__( else: model_path = download_model( model_size_or_path, + output_dir, local_files_only=local_files_only, cache_dir=download_root, )