qurator-spk/eynollah
1
0
Fork 0
mirror of https://github.com/qurator-spk/eynollah.git synced 2025-10-15 19:09:58 +02:00
Code Issues Projects Releases Wiki Activity

setup_models: avoid unnecessarily loading region_fl

Browse source
This commit is contained in:
Robert Sachunsky 2025-10-14 14:23:29 +02:00
parent e8b7212f36
commit 8299e7009a
1 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

6
src/eynollah/eynollah.py
View file

@ -385,6 +385,8 @@ class Eynollah:
self.logger.warning("overriding default model %s version %s to %s", key, self.model_versions[key], val) self.logger.warning("overriding default model %s version %s to %s", key, self.model_versions[key], val)
self.model_versions[key] = val self.model_versions[key] = val
# load models, depending on modes # load models, depending on modes
# (note: loading too many models can cause OOM on GPU/CUDA,
# thus, we try set up the minimal configuration for the current mode)
loadable = [ loadable = [
"col_classifier", "col_classifier",
"binarization", "binarization",
@ -400,8 +402,8 @@ class Eynollah:
# if self.allow_enhancement:? # if self.allow_enhancement:?
loadable.append("enhancement") loadable.append("enhancement")
if self.full_layout: if self.full_layout:
loadable.extend(["region_fl_np", loadable.append("region_fl_np")
"region_fl"]) #loadable.append("region_fl")
if self.reading_order_machine_based: if self.reading_order_machine_based:
loadable.append("reading_order") loadable.append("reading_order")
if self.tables: if self.tables: