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] 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