diff --git a/whisper/__init__.py b/whisper/__init__.py index e210718f3..6f8735294 100644 --- a/whisper/__init__.py +++ b/whisper/__init__.py @@ -103,7 +103,7 @@ def available_models() -> List[str]: def load_model( name: str, device: Optional[Union[str, torch.device]] = None, - download_root: str = None, + download_root: Optional[str] = None, in_memory: bool = False, ) -> Whisper: """