mirrors/docling
1
0
Fork 0
mirror of https://github.com/DS4SD/docling.git synced 2025-07-25 19:44:34 +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
parent 67bf4d47ba
commit b552fec0c6
2 changed files with 2 additions and 2 deletions
Show Stats 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, local_dir: Optional[Path] = None,
force: bool = False, force: bool = False,
progress: bool = False, progress: bool = False,
layout_model_config: LayoutModelConfig = DOCLING_LAYOUT_V2, layout_model_config: LayoutModelConfig = LayoutOptions().model_spec, # use default
) -> Path: ) -> Path:
return download_hf_model( return download_hf_model(
repo_id=layout_model_config.repo_id, 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: if with_layout:
_log.info("Downloading layout model...") _log.info("Downloading layout model...")
LayoutModel.download_models( 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, force=force,
progress=progress, progress=progress,
) )