ModelZoo: add ocr key for memory_limit

This commit is contained in:
Robert Sachunsky 2026-05-21 22:50:13 +02:00
parent ea41dcae1d
commit f3a93983c0

View file

@ -185,6 +185,7 @@ class EynollahModelZoo:
"region_fl_np": 1756, "region_fl_np": 1756,
"table": 1818, "table": 1818,
"reading_order": 632, "reading_order": 632,
"ocr": 850,
}[model_category])]) }[model_category])])
vendor_name = ( vendor_name = (
tf.config.experimental.get_device_details(device) tf.config.experimental.get_device_details(device)