mirror of
https://github.com/openai/whisper.git
synced 2025-03-30 14:28:27 +00:00
fix type hint for load_model function
This commit is contained in:
parent
517a43ecd1
commit
310abe10a2
@ -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:
|
||||
"""
|
||||
|
Loading…
x
Reference in New Issue
Block a user