Mirroristas/whisper
2
0
Fork 0
mirror of https://github.com/openai/whisper.git synced 2025-11-23 22:15:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

Updated Version parsing for torch

Browse Source 
Since pkg_resources has been deprecated from python 3.12, we have to use 'packaging' to handle the version parsing instead.
This commit is contained in:
R I P Skillan 2025-09-27 07:02:09 +05:30 committed by GitHub
parent ea61a1757f
commit bf24f74a34
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
whisper/__init__.py
View File

@ -4,6 +4,7 @@ import os
import urllib import urllib
import warnings import warnings
from typing import List, Optional, Union from typing import List, Optional, Union
from packaging import version
import torch import torch
from tqdm import tqdm from tqdm import tqdm
@ -13,7 +14,6 @@ from .decoding import DecodingOptions, DecodingResult, decode, detect_language
from .model import ModelDimensions, Whisper from .model import ModelDimensions, Whisper
from .transcribe import transcribe from .transcribe import transcribe
from .version import __version__ from .version import __version__
from pkg_resources import packaging
_MODELS = { _MODELS = {
"tiny.en": "https://openaipublic.azureedge.net/main/whisper/models/d3dd57d32accea0b295c96e26691aa14d8822fac7d9d27d5dc00b4ca2826dd03/tiny.en.pt", "tiny.en": "https://openaipublic.azureedge.net/main/whisper/models/d3dd57d32accea0b295c96e26691aa14d8822fac7d9d27d5dc00b4ca2826dd03/tiny.en.pt",
@ -148,7 +148,7 @@ def load_model(
with ( with (
io.BytesIO(checkpoint_file) if in_memory else open(checkpoint_file, "rb") io.BytesIO(checkpoint_file) if in_memory else open(checkpoint_file, "rb")
) as fp: ) as fp:
kwargs = {"weights_only": True} if packaging.version.parse(torch.__version__ ) >= packaging.version.parse("1.13") else {} kwargs = {"weights_only": True} if version.parse(torch.__version__ ) >= version.parse("1.13") else {}
checkpoint = torch.load(fp, map_location=device, **kwargs) checkpoint = torch.load(fp, map_location=device, **kwargs)
del checkpoint_file del checkpoint_file