docling-project/docling-ibm-models
1
0
Fork 0
mirror of https://github.com/docling-project/docling-ibm-models.git synced 2026-05-17 13:10:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: load code formula model with weights_only=True (#158)

Browse Source 
Signed-off-by: Panos Vagenas <pva@zurich.ibm.com>
This commit is contained in:
Panos Vagenas
2026-04-23 09:33:03 +02:00
committed by GitHub
parent 6d4043c67c
commit 2ebbcad084
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
docling_ibm_models/code_formula_model/models/sam.py
+1 -1
View File
@@ -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