From 05c8cb0fba524da3db6762f6b9eb1d1392c992b6 Mon Sep 17 00:00:00 2001 From: Christoph Auer Date: Tue, 10 Dec 2024 20:02:19 +0100 Subject: [PATCH] Update HF model ref, reset test generate Signed-off-by: Christoph Auer Signed-off-by: Christoph Auer --- docling/pipeline/standard_pdf_pipeline.py | 2 +- tests/test_e2e_conversion.py | 4 ++-- tests/test_e2e_ocr_conversion.py | 4 ++-- 3 files changed, 5 insertions(+), 5 deletions(-) diff --git a/docling/pipeline/standard_pdf_pipeline.py b/docling/pipeline/standard_pdf_pipeline.py index 68a4dcf2..2f8c1421 100644 --- a/docling/pipeline/standard_pdf_pipeline.py +++ b/docling/pipeline/standard_pdf_pipeline.py @@ -106,7 +106,7 @@ class StandardPdfPipeline(PaginatedPipeline): repo_id="ds4sd/docling-models", force_download=force, local_dir=local_dir, - revision="refs/pr/2", + revision="v2.1.0", ) return Path(download_path) diff --git a/tests/test_e2e_conversion.py b/tests/test_e2e_conversion.py index 77d3d07a..d6753b04 100644 --- a/tests/test_e2e_conversion.py +++ b/tests/test_e2e_conversion.py @@ -8,8 +8,8 @@ from docling.document_converter import DocumentConverter, PdfFormatOption from .verify_utils import verify_conversion_result_v1, verify_conversion_result_v2 -GENERATE_V1 = True -GENERATE_V2 = True +GENERATE_V1 = False +GENERATE_V2 = False def get_pdf_paths(): diff --git a/tests/test_e2e_ocr_conversion.py b/tests/test_e2e_ocr_conversion.py index fe9bb9cf..73a943af 100644 --- a/tests/test_e2e_ocr_conversion.py +++ b/tests/test_e2e_ocr_conversion.py @@ -18,8 +18,8 @@ from docling.document_converter import DocumentConverter, PdfFormatOption from .verify_utils import verify_conversion_result_v1, verify_conversion_result_v2 -GENERATE_V1 = True -GENERATE_V2 = True +GENERATE_V1 = False +GENERATE_V2 = False def get_pdf_paths():