From 14a7b3d0868749125bb24b4dae1c29265d800be0 Mon Sep 17 00:00:00 2001 From: Christoph Auer Date: Wed, 23 Jul 2025 08:51:07 +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 | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/docling/models/layout_model.py b/docling/models/layout_model.py index 256c2965..76fc4ec1 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 = LayoutOptions().model_spec, # use default + 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 43fff7ed..28c3918c 100644 --- a/docling/utils/model_downloader.py +++ b/docling/utils/model_downloader.py @@ -4,6 +4,7 @@ from typing import Optional from docling.datamodel.layout_model_specs import DOCLING_LAYOUT_V2 from docling.datamodel.pipeline_options import ( + LayoutOptions, granite_picture_description, smolvlm_picture_description, )