From 12fc5d76e7a43ff9ba0f36d3bba88de893cf1971 Mon Sep 17 00:00:00 2001 From: Thien Dang <110881337+tnt305@users.noreply.github.com> Date: Mon, 17 Feb 2025 09:34:21 +0700 Subject: [PATCH] Update transcribe.py --- faster_whisper/transcribe.py | 2 ++ 1 file changed, 2 insertions(+) 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, )