docling-ibm-models 1.1.1__tar.gz → 1.1.2__tar.gz

{docling_ibm_models-1.1.1 → docling_ibm_models-1.1.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: docling-ibm-models
-Version: 1.1.1
+Version: 1.1.2
 Summary: This package contains the AI models used by the Docling PDF conversion package
 License: MIT
 Keywords: docling,convert,document,pdf,layout model,segmentation,table structure,table former
@@ -26,8 +26,7 @@ Requires-Dist: lxml (>=4.9.1,<5.0.0)
 Requires-Dist: mean_average_precision (>=2021.4.26.0,<2022.0.0.0)
 Requires-Dist: numpy (>=1.24.4,<2.0.0)
 Requires-Dist: onnxruntime (>=1.16.2,<2.0.0)
-Requires-Dist: opencv-python (>=4.9.0.80,<5.0.0.0) ; sys_platform != "linux"
-Requires-Dist: opencv-python-headless (>=4.9.0.80,<5.0.0.0) ; sys_platform == "linux"
+Requires-Dist: opencv-python-headless (>=4.9.0.80,<5.0.0.0)
 Requires-Dist: torch (>=2.2.2,<3.0.0)
 Requires-Dist: torchvision (>=0.17.2)
 Requires-Dist: tqdm (>=4.64.0,<5.0.0)

{docling_ibm_models-1.1.1 → docling_ibm_models-1.1.2}/docling_ibm_models/tableformer/data_management/tf_predictor.py

@@ -99,8 +99,13 @@ class TFPredictor:
     Table predictions for the in-memory Docling API
     """
 
-    def __init__(self, config):
+    def __init__(self, config, num_threads: int = None):
         r"""
+        The number of threads is decided, in the following order, by:
+        1. The init method parameter `num_threads`, if it is set.
+        2. The envvar "OMP_NUM_THREADS", if it is set.
+        3. The default value 4.
+
         Parameters
         ----------
         config : dict
@@ -123,6 +128,13 @@ class TFPredictor:
         self._post_processor = MatchingPostProcessor(config)
 
         self._init_word_map()
+
+        # Set the number of torch threads
+        if num_threads is None:
+            num_threads = int(os.environ.get("OMP_NUM_THREADS", 4))
+        self._num_threads = num_threads
+        torch.set_num_threads(num_threads)
+
         # Load the model
         self._model = self._load_model()
         self._model.eval()

{docling_ibm_models-1.1.1 → docling_ibm_models-1.1.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "docling-ibm-models"
-version = "1.1.1"  # DO NOT EDIT, updated automatically
+version = "1.1.2"  # DO NOT EDIT, updated automatically
 description = "This package contains the AI models used by the Docling PDF conversion package"
 authors = ["Nikos Livathinos <nli@zurich.ibm.com>", "Maxim Lysak <mly@zurich.ibm.com>", "Ahmed Nassar <ahn@zurich.ibm.com>", "Christoph Auer <cau@zurich.ibm.com>", "Michele Dolfi <dol@zurich.ibm.com>", "Peter Staar <taa@zurich.ibm.com>"]
 license = "MIT"
@@ -33,8 +33,7 @@ tqdm = "^4.64.0"
 apted = "^1.0.3"
 Distance = "^0.1.3"
 mean_average_precision = "^2021.4.26.0"
-opencv-python-headless = { version = "^4.9.0.80", markers = 'sys_platform=="linux"'}
-opencv-python = { version = "^4.9.0.80", markers = 'sys_platform!="linux"'}
+opencv-python-headless = { version = "^4.9.0.80" }
 
 [tool.poetry.dev-dependencies]
 black = {extras = ["jupyter"], version = "^24.4.2"}