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:51:07 +02:00
parent b552fec0c6
commit 14a7b3d086
2 changed files with 2 additions and 1 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 = LayoutOptions().model_spec, # use default 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,

1
docling/utils/model_downloader.py
View File

@ -4,6 +4,7 @@ from typing import Optional
from docling.datamodel.layout_model_specs import DOCLING_LAYOUT_V2 from docling.datamodel.layout_model_specs import DOCLING_LAYOUT_V2
from docling.datamodel.pipeline_options import ( from docling.datamodel.pipeline_options import (
LayoutOptions,
granite_picture_description, granite_picture_description,
smolvlm_picture_description, smolvlm_picture_description,
) )