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, )