Mirroristas/whisper
2
0
Fork 0
mirror of https://github.com/openai/whisper.git synced 2025-11-24 14:35:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update __init__.py

Browse Source
This commit is contained in:
Jong Wook Kim 2025-06-25 17:45:54 -07:00 committed by GitHub
parent 8c3f996e1e
commit 915b3e747b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
whisper/__init__.py
View File

@ -147,7 +147,10 @@ def load_model(
with (
io.BytesIO(checkpoint_file) if in_memory else open(checkpoint_file, "rb")
) as fp:
checkpoint = torch.load(fp, map_location=device, weights_only=True)
try:
checkpoint = torch.load(fp, map_location=device, weights_only=True)
except TypeError: # for compatibility with older torch
checkpoint = torch.load(fp, map_location=device)
del checkpoint_file
dims = ModelDimensions(**checkpoint["dims"])