From 1963245ed15a9f232ac3b763c7a03351d77ae799 Mon Sep 17 00:00:00 2001 From: Russell Bryant Date: Fri, 9 Jan 2026 19:28:57 -0500 Subject: [PATCH] [Core] Use weights_only=True with torch.load (#32045) Signed-off-by: Russell Bryant --- vllm/model_executor/model_loader/tensorizer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/vllm/model_executor/model_loader/tensorizer.py b/vllm/model_executor/model_loader/tensorizer.py index e4e530f0c..757c467c4 100644 --- a/vllm/model_executor/model_loader/tensorizer.py +++ b/vllm/model_executor/model_loader/tensorizer.py @@ -762,7 +762,7 @@ def tensorize_lora_adapter(lora_path: str, tensorizer_config: TensorizerConfig): if tensor_path.endswith(".safetensors"): tensors = safetensors.torch.load_file(tensor_path) elif tensor_path.endswith(".bin"): - tensors = torch.load(tensor_path) + tensors = torch.load(tensor_path, weights_only=True) else: raise ValueError("Unsupported file: %s", tensor_path)