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():