From 2ebbcad084e5df031d1835638bf865b0a0af22c0 Mon Sep 17 00:00:00 2001 From: Panos Vagenas <35837085+vagenas@users.noreply.github.com> Date: Thu, 23 Apr 2026 09:33:03 +0200 Subject: [PATCH] fix: load code formula model with `weights_only=True` (#158) Signed-off-by: Panos Vagenas --- docling_ibm_models/code_formula_model/models/sam.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docling_ibm_models/code_formula_model/models/sam.py b/docling_ibm_models/code_formula_model/models/sam.py index 98179cf..c5b0f96 100644 --- a/docling_ibm_models/code_formula_model/models/sam.py +++ b/docling_ibm_models/code_formula_model/models/sam.py @@ -508,7 +508,7 @@ def _build_sam( if checkpoint is not None: # with open(checkpoint, "rb") as f: - state_dict = torch.load(checkpoint) + state_dict = torch.load(checkpoint, weights_only=True) image_encoder.load_state_dict(state_dict, strict=True) return image_encoder