mirrors/docling
1
0
Fork 0
mirror of https://github.com/DS4SD/docling.git synced 2025-12-08 12:48:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use default layout model in model_downloader default args

Browse Source 
Signed-off-by: Christoph Auer <cau@zurich.ibm.com>
This commit is contained in:
Christoph Auer
2025-07-23 08:50:22 +02:00
committed by Nikos Livathinos
parent a982995fb7
commit e0482723c4
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
docling/models/layout_model.py
View File

@@ -91,7 +91,7 @@ class LayoutModel(BasePageModel):
local_dir: Optional[Path] = None,
force: bool = False,
progress: bool = False,
layout_model_config: LayoutModelConfig = DOCLING_LAYOUT_V2,
layout_model_config: LayoutModelConfig = LayoutOptions().model_spec, # use default
) -> Path:
return download_hf_model(
repo_id=layout_model_config.repo_id,

2
docling/utils/model_downloader.py
View File

@@ -47,7 +47,7 @@ def download_models(
if with_layout:
_log.info("Downloading layout model...")
LayoutModel.download_models(
local_dir=output_dir / DOCLING_LAYOUT_V2.model_repo_folder,
local_dir=output_dir / LayoutOptions().model_spec.model_repo_folder,
force=force,
progress=progress,
)