From 310abe10a29e8f2585fd9226bffdd5fff2ae9d23 Mon Sep 17 00:00:00 2001 From: Stanley F Date: Fri, 31 Jan 2025 14:08:32 +0100 Subject: [PATCH] fix type hint for load_model function --- whisper/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/whisper/__init__.py b/whisper/__init__.py index e210718..6f87352 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: """