From 6f9d68dd6b94da6d3e84ce631f85804aadf1c24d Mon Sep 17 00:00:00 2001 From: Guillaume Klein Date: Wed, 26 Apr 2023 17:36:24 +0200 Subject: [PATCH] Fix typing of local_files_only --- faster_whisper/transcribe.py | 2 +- faster_whisper/utils.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/faster_whisper/transcribe.py b/faster_whisper/transcribe.py index 4d7f0f6..3973fe6 100644 --- a/faster_whisper/transcribe.py +++ b/faster_whisper/transcribe.py @@ -79,7 +79,7 @@ class WhisperModel: cpu_threads: int = 0, num_workers: int = 1, download_root: Optional[str] = None, - local_files_only: Optional[bool] = False, + local_files_only: bool = False, ): """Initializes the Whisper model. diff --git a/faster_whisper/utils.py b/faster_whisper/utils.py index 34a310a..a052ad8 100644 --- a/faster_whisper/utils.py +++ b/faster_whisper/utils.py @@ -34,7 +34,7 @@ def get_logger(): def download_model( size: str, output_dir: Optional[str] = None, - local_files_only: Optional[bool] = False, + local_files_only: bool = False, cache_dir: Optional[str] = None, ): """Downloads a CTranslate2 Whisper model from the Hugging Face Hub.