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 warnings
from typing import List, Optional, Union
from packaging import version
import torch
from tqdm import tqdm
@ -13,7 +14,6 @@ from .decoding import DecodingOptions, DecodingResult, decode, detect_language
from .model import ModelDimensions, Whisper
from .transcribe import transcribe
from .version import __version__
from pkg_resources import packaging
_MODELS = {
"tiny.en": "https://openaipublic.azureedge.net/main/whisper/models/d3dd57d32accea0b295c96e26691aa14d8822fac7d9d27d5dc00b4ca2826dd03/tiny.en.pt",
@ -148,7 +148,7 @@ def load_model(
with (
io.BytesIO(checkpoint_file) if in_memory else open(checkpoint_file, "rb")
) 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)
del checkpoint_file