fix type hint for load_model function

This commit is contained in:
Stanley F 2025-01-31 14:08:32 +01:00 committed by GitHub
parent 517a43ecd1
commit 310abe10a2
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -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:
"""