Docling/tests
Peter W. J. Staar cfdf4cea25
feat: new vlm-models support (#1570)
* feat: adding new vlm-models support

Signed-off-by: Peter Staar <taa@zurich.ibm.com>

* fixed the transformers

Signed-off-by: Peter Staar <taa@zurich.ibm.com>

* got microsoft/Phi-4-multimodal-instruct to work

Signed-off-by: Peter Staar <taa@zurich.ibm.com>

* working on vlm's

Signed-off-by: Peter Staar <taa@zurich.ibm.com>

* refactoring the VLM part

Signed-off-by: Peter Staar <taa@zurich.ibm.com>

* all working, now serious refacgtoring necessary

Signed-off-by: Peter Staar <taa@zurich.ibm.com>

* refactoring the download_model

Signed-off-by: Peter Staar <taa@zurich.ibm.com>

* added the formulate_prompt

Signed-off-by: Peter Staar <taa@zurich.ibm.com>

* pixtral 12b runs via MLX and native transformers

Signed-off-by: Peter Staar <taa@zurich.ibm.com>

* added the VlmPredictionToken

Signed-off-by: Peter Staar <taa@zurich.ibm.com>

* refactoring minimal_vlm_pipeline

Signed-off-by: Peter Staar <taa@zurich.ibm.com>

* fixed the MyPy

Signed-off-by: Peter Staar <taa@zurich.ibm.com>

* added pipeline_model_specializations file

Signed-off-by: Peter Staar <taa@zurich.ibm.com>

* need to get Phi4 working again ...

Signed-off-by: Peter Staar <taa@zurich.ibm.com>

* finalising last points for vlms support

Signed-off-by: Peter Staar <taa@zurich.ibm.com>

* fixed the pipeline for Phi4

Signed-off-by: Peter Staar <taa@zurich.ibm.com>

* streamlining all code

Signed-off-by: Peter Staar <taa@zurich.ibm.com>

* reformatted the code

Signed-off-by: Peter Staar <taa@zurich.ibm.com>

* fixing the tests

Signed-off-by: Peter Staar <taa@zurich.ibm.com>

* added the html backend to the VLM pipeline

Signed-off-by: Peter Staar <taa@zurich.ibm.com>

* fixed the static load_from_doctags

Signed-off-by: Peter Staar <taa@zurich.ibm.com>

* restore stable imports

Signed-off-by: Michele Dolfi <dol@zurich.ibm.com>

* use AutoModelForVision2Seq for Pixtral and review example (including rename)

Signed-off-by: Michele Dolfi <dol@zurich.ibm.com>

* remove unused value

Signed-off-by: Michele Dolfi <dol@zurich.ibm.com>

* refactor instances of VLM models

Signed-off-by: Michele Dolfi <dol@zurich.ibm.com>

* skip compare example in CI

Signed-off-by: Michele Dolfi <dol@zurich.ibm.com>

* use lowercase and uppercase only

Signed-off-by: Michele Dolfi <dol@zurich.ibm.com>

* add new minimal_vlm example and refactor pipeline_options_vlm_model for cleaner import

Signed-off-by: Michele Dolfi <dol@zurich.ibm.com>

* rename pipeline_vlm_model_spec

Signed-off-by: Michele Dolfi <dol@zurich.ibm.com>

* move more argument to options and simplify model init

Signed-off-by: Michele Dolfi <dol@zurich.ibm.com>

* add supported_devices

Signed-off-by: Michele Dolfi <dol@zurich.ibm.com>

* remove not-needed function

Signed-off-by: Michele Dolfi <dol@zurich.ibm.com>

* exclude minimal_vlm

Signed-off-by: Michele Dolfi <dol@zurich.ibm.com>

* missing file

Signed-off-by: Michele Dolfi <dol@zurich.ibm.com>

* add message for transformers version

Signed-off-by: Michele Dolfi <dol@zurich.ibm.com>

* rename to specs

Signed-off-by: Michele Dolfi <dol@zurich.ibm.com>

* use module import and remove MLX from non-darwin

Signed-off-by: Michele Dolfi <dol@zurich.ibm.com>

* remove hf_vlm_model and add extra_generation_args

Signed-off-by: Michele Dolfi <dol@zurich.ibm.com>

* use single HF VLM model class

Signed-off-by: Michele Dolfi <dol@zurich.ibm.com>

* remove torch type

Signed-off-by: Michele Dolfi <dol@zurich.ibm.com>

* add docs for vision models

Signed-off-by: Michele Dolfi <dol@zurich.ibm.com>

---------

Signed-off-by: Peter Staar <taa@zurich.ibm.com>
Signed-off-by: Michele Dolfi <dol@zurich.ibm.com>
Co-authored-by: Michele Dolfi <dol@zurich.ibm.com>
2025-06-02 17:01:06 +02:00
..
data test: add missing ground truth files (#1667) 2025-05-28 13:26:49 +02:00
data_scanned feat(ocr): auto-detect rotated pages in Tesseract (#1167) 2025-05-21 18:12:33 +02:00
__init__.py fix: Add unit tests (#51) 2024-08-30 14:08:20 +02:00
test_backend_asciidoc.py ci: add coverage and ruff (#1383) 2025-04-14 18:01:26 +02:00
test_backend_csv.py chore: fix or ignore runtime and deprecation warnings (#1660) 2025-05-28 17:55:31 +02:00
test_backend_docling_json.py feat: add Docling JSON ingestion (#783) 2025-01-24 18:05:23 +01:00
test_backend_docling_parse_v2.py ci: add coverage and ruff (#1383) 2025-04-14 18:01:26 +02:00
test_backend_docling_parse_v4.py ci: add coverage and ruff (#1383) 2025-04-14 18:01:26 +02:00
test_backend_docling_parse.py ci: add coverage and ruff (#1383) 2025-04-14 18:01:26 +02:00
test_backend_html.py ci: add coverage and ruff (#1383) 2025-04-14 18:01:26 +02:00
test_backend_jats.py ci: add coverage and ruff (#1383) 2025-04-14 18:01:26 +02:00
test_backend_markdown.py fix(markdown): handle nested lists (#910) 2025-02-07 12:55:12 +01:00
test_backend_msexcel.py ci: add coverage and ruff (#1383) 2025-04-14 18:01:26 +02:00
test_backend_msword.py feat: add textbox content extraction in msword_backend (#1538) 2025-05-19 15:01:36 +02:00
test_backend_patent_uspto.py ci: add coverage and ruff (#1383) 2025-04-14 18:01:26 +02:00
test_backend_pdfium.py fix(pypdfium): resolve overlapping text when merging bounding boxes (#1549) 2025-05-19 15:26:00 +02:00
test_backend_pptx.py ci: add coverage and ruff (#1383) 2025-04-14 18:01:26 +02:00
test_backend_webp.py feat: support image/webp file type (#1415) 2025-05-14 09:47:28 +02:00
test_cli.py fix: Test cases for RTL programmatic PDFs and fixes for the formula model (#903) 2025-02-07 08:43:31 +01:00
test_code_formula.py ci: add coverage and ruff (#1383) 2025-04-14 18:01:26 +02:00
test_data_gen_flag.py fix(markdown): handle nested lists (#910) 2025-02-07 12:55:12 +01:00
test_document_picture_classifier.py ci: add coverage and ruff (#1383) 2025-04-14 18:01:26 +02:00
test_e2e_conversion.py feat: new vlm-models support (#1570) 2025-06-02 17:01:06 +02:00
test_e2e_ocr_conversion.py feat: new vlm-models support (#1570) 2025-06-02 17:01:06 +02:00
test_input_doc.py fix: guess HTML content starting with script tag (#1673) 2025-06-02 08:43:24 +02:00
test_interfaces.py ci: add coverage and ruff (#1383) 2025-04-14 18:01:26 +02:00
test_invalid_input.py ci: add coverage and ruff (#1383) 2025-04-14 18:01:26 +02:00
test_legacy_format_transform.py chore: fix or ignore runtime and deprecation warnings (#1660) 2025-05-28 17:55:31 +02:00
test_options.py feat: new vlm-models support (#1570) 2025-06-02 17:01:06 +02:00
test_settings_load.py fix(settings): fix nested settings load via environment variables (#1551) 2025-05-14 13:42:10 +02:00
verify_utils.py test: ensure utf-8 in test data utils (#1691) 2025-06-02 12:13:19 +02:00