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