From b552fec0c6440be4f891ec329e95d7b13660a3f0 Mon Sep 17 00:00:00 2001 From: Christoph Auer Date: Wed, 23 Jul 2025 08:50:22 +0200 Subject: [PATCH] Use default layout model in model_downloader default args Signed-off-by: Christoph Auer --- docling/models/layout_model.py | 2 +- docling/utils/model_downloader.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/docling/models/layout_model.py b/docling/models/layout_model.py index 05a86f31..256c2965 100644 --- a/docling/models/layout_model.py +++ b/docling/models/layout_model.py @@ -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, diff --git a/docling/utils/model_downloader.py b/docling/utils/model_downloader.py index a2994fb7..43fff7ed 100644 --- a/docling/utils/model_downloader.py +++ b/docling/utils/model_downloader.py @@ -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, )