From f53acf2de6cc928501d84b764eb67d9255a89cc1 Mon Sep 17 00:00:00 2001 From: R I P Skillan <123875500+RIP-Skillan@users.noreply.github.com> Date: Tue, 23 Sep 2025 22:10:41 +0530 Subject: [PATCH 1/3] Fix for pytorch version check issue Fix for Pytorch version check issue for adding 'weights_only' argument. Added a more accurate version parser. --- whisper/__init__.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/whisper/__init__.py b/whisper/__init__.py index f284ec0..f7850cd 100644 --- a/whisper/__init__.py +++ b/whisper/__init__.py @@ -13,6 +13,7 @@ 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", @@ -147,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 torch.__version__ >= "1.13" else {} + kwargs = {"weights_only": True} if packaging.version.parse(torch.__version__ ) >= packaging.version.parse("1.13") else {} checkpoint = torch.load(fp, map_location=device, **kwargs) del checkpoint_file From ea61a1757f18a37ba2f64cf2914eecaae04ed6d5 Mon Sep 17 00:00:00 2001 From: R I P Skillan <123875500+RIP-Skillan@users.noreply.github.com> Date: Sat, 27 Sep 2025 06:57:44 +0530 Subject: [PATCH 2/3] Added packaging for version parsing version parsing which could originally be done using pkg_resources is deprecated from python 3.12, therefore requiring packaging to handle version parsing --- requirements.txt | 1 + 1 file changed, 1 insertion(+) diff --git a/requirements.txt b/requirements.txt index 8ee5920..9aea844 100644 --- a/requirements.txt +++ b/requirements.txt @@ -5,3 +5,4 @@ tqdm more-itertools tiktoken triton>=2.0.0;platform_machine=="x86_64" and sys_platform=="linux" or sys_platform=="linux2" +packaging From bf24f74a349445487d53f58cbacc8d17fdb10c38 Mon Sep 17 00:00:00 2001 From: R I P Skillan <123875500+RIP-Skillan@users.noreply.github.com> Date: Sat, 27 Sep 2025 07:02:09 +0530 Subject: [PATCH 3/3] Updated Version parsing for torch Since pkg_resources has been deprecated from python 3.12, we have to use 'packaging' to handle the version parsing instead. --- whisper/__init__.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/whisper/__init__.py b/whisper/__init__.py index f7850cd..cae01b3 100644 --- a/whisper/__init__.py +++ b/whisper/__init__.py @@ -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