Update HF model ref, reset test generate

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

Signed-off-by: Christoph Auer <cau@zurich.ibm.com>
This commit is contained in:
Christoph Auer 2024-12-10 20:02:19 +01:00
parent 1de42bef6a
commit 05c8cb0fba
3 changed files with 5 additions and 5 deletions

View File

@ -106,7 +106,7 @@ class StandardPdfPipeline(PaginatedPipeline):
repo_id="ds4sd/docling-models", repo_id="ds4sd/docling-models",
force_download=force, force_download=force,
local_dir=local_dir, local_dir=local_dir,
revision="refs/pr/2", revision="v2.1.0",
) )
return Path(download_path) return Path(download_path)

View File

@ -8,8 +8,8 @@ from docling.document_converter import DocumentConverter, PdfFormatOption
from .verify_utils import verify_conversion_result_v1, verify_conversion_result_v2 from .verify_utils import verify_conversion_result_v1, verify_conversion_result_v2
GENERATE_V1 = True GENERATE_V1 = False
GENERATE_V2 = True GENERATE_V2 = False
def get_pdf_paths(): def get_pdf_paths():

View File

@ -18,8 +18,8 @@ from docling.document_converter import DocumentConverter, PdfFormatOption
from .verify_utils import verify_conversion_result_v1, verify_conversion_result_v2 from .verify_utils import verify_conversion_result_v1, verify_conversion_result_v2
GENERATE_V1 = True GENERATE_V1 = False
GENERATE_V2 = True GENERATE_V2 = False
def get_pdf_paths(): def get_pdf_paths():