Fix typing of local_files_only

This commit is contained in:
Guillaume Klein
2023-04-26 17:36:24 +02:00
parent 68df3214ba
commit 6f9d68dd6b
2 changed files with 2 additions and 2 deletions

View File

@@ -79,7 +79,7 @@ class WhisperModel:
cpu_threads: int = 0, cpu_threads: int = 0,
num_workers: int = 1, num_workers: int = 1,
download_root: Optional[str] = None, download_root: Optional[str] = None,
local_files_only: Optional[bool] = False, local_files_only: bool = False,
): ):
"""Initializes the Whisper model. """Initializes the Whisper model.

View File

@@ -34,7 +34,7 @@ def get_logger():
def download_model( def download_model(
size: str, size: str,
output_dir: Optional[str] = None, output_dir: Optional[str] = None,
local_files_only: Optional[bool] = False, local_files_only: bool = False,
cache_dir: Optional[str] = None, cache_dir: Optional[str] = None,
): ):
"""Downloads a CTranslate2 Whisper model from the Hugging Face Hub. """Downloads a CTranslate2 Whisper model from the Hugging Face Hub.