Update to latest HF models

Signed-off-by: Christoph Auer <cau@zurich.ibm.com>
This commit is contained in:
Christoph Auer 2024-10-08 18:54:15 +02:00
parent dd8a0e9e44
commit a8cada3694
2 changed files with 2 additions and 2 deletions

View File

@ -35,7 +35,7 @@ _log = logging.getLogger(__name__)
layout_label_to_ds_type = { layout_label_to_ds_type = {
"Title": "title", "Title": "title",
"Document Index": "table-of-contents", "Document Index": "table", # table-of-contents",
"Section-header": "subtitle-level-1", "Section-header": "subtitle-level-1",
"Checkbox-Selected": "checkbox-selected", "Checkbox-Selected": "checkbox-selected",
"Checkbox-Unselected": "checkbox-unselected", "Checkbox-Unselected": "checkbox-unselected",

View File

@ -70,7 +70,7 @@ class DocumentConverter:
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="v2.0.0", revision="v2.0.1",
) )
return Path(download_path) return Path(download_path)