search

DS4SD / docling-ibm-models

MIT License
45 stars 10 forks source link

feat: Introduce logic to control the number of threads in TFPredictor #11

Closed nikos-livathinos closed 3 months ago

nikos-livathinos commented 3 months ago

Control the number of torch intraop threads in TFPredictor with the one defined first:

  1. The init parameter num_threads.
  2. The envvar OMP_NUM_THREADS.
  3. A default value.