docling/docling/datamodel
Ahmed Nassar 77eb77bdc2
feat: Support cuda:n GPU device allocation (#694)
* Adding multi-gpu support, and cuda device allocation

Signed-off-by: ahn <ahn@zurich.ibm.com>

* Fixes pydantic exception with cuda:n
Signed-off-by: ahn <ahn@zurich.ibm.com>

* Pydantic field validator and comment restored.

Signed-off-by: ahn <ahn@zurich.ibm.com>

* chore: Accept AcceleratorDevice enum type

Signed-off-by: Christoph Auer <cau@zurich.ibm.com>

* Resetted some options to default, removed EasyOCR model wrap.
Signed-off-by: ahn <ahn@zurich.ibm.com>

* Fixed rebased issues
Signed-off-by: ahn <ahn@zurich.ibm.com>

* Revert accelerator test options
Signed-off-by: ahn <ahn@zurich.ibm.com>

---------

Signed-off-by: ahn <ahn@zurich.ibm.com>
Signed-off-by: Christoph Auer <cau@zurich.ibm.com>
Co-authored-by: ahn <ahn@sonny.zuvela.ibm.com>
Co-authored-by: ahn <ahn@zurich.ibm.com>
Co-authored-by: Christoph Auer <cau@zurich.ibm.com>
2025-02-17 11:31:13 +01:00
..
__init__.py Initial commit 2024-07-15 09:42:42 +02:00
base_models.py feat(xml-jats): parse XML JATS documents (#967) 2025-02-17 10:43:31 +01:00
document.py feat(xml-jats): parse XML JATS documents (#967) 2025-02-17 10:43:31 +01:00
pipeline_options.py feat: Support cuda:n GPU device allocation (#694) 2025-02-17 11:31:13 +01:00
settings.py feat: allow artifacts_path to be defined as ENV (#940) 2025-02-12 13:08:37 +01:00