From 83876b4bc76eedf8b9ee76f098a70178a50080d3 Mon Sep 17 00:00:00 2001 From: yaslack Date: Wed, 20 Nov 2024 18:05:08 +0100 Subject: [PATCH] Fix: Update torch.load to use weights_only=True to prevent security warning --- whisper/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/whisper/__init__.py b/whisper/__init__.py index e210718f3..e6798839b 100644 --- a/whisper/__init__.py +++ b/whisper/__init__.py @@ -147,7 +147,7 @@ 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) + checkpoint = torch.load(fp, map_location=device, weights_only=True) del checkpoint_file dims = ModelDimensions(**checkpoint["dims"])