diff --git a/docling_ibm_models/tableformer/models/common/base_model.py b/docling_ibm_models/tableformer/models/common/base_model.py
index 2ed6ff8..2bda350 100644
--- a/docling_ibm_models/tableformer/models/common/base_model.py
+++ b/docling_ibm_models/tableformer/models/common/base_model.py
@@ -254,7 +254,9 @@ def _load_best_checkpoint(self):
                     "Loading model checkpoint file: {}".format(checkpoint_file)
                 )
                 saved_model = torch.load(
-                    checkpoint_file, map_location=self._device, weights_only=False
+                    checkpoint_file,
+                    map_location=self._device,
+                    weights_only=True,
                 )
                 return saved_model, checkpoint_file
             except RuntimeError: