From 0867936a3aa5b53df997063cb61bee741b3f5aad Mon Sep 17 00:00:00 2001 From: Saidgurbuz Date: Tue, 20 May 2025 09:42:34 +0200 Subject: [PATCH] fix load_from_doctags usage Signed-off-by: Saidgurbuz --- docling/pipeline/vlm_pipeline.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/docling/pipeline/vlm_pipeline.py b/docling/pipeline/vlm_pipeline.py index 9a7b51eb..0e51332c 100644 --- a/docling/pipeline/vlm_pipeline.py +++ b/docling/pipeline/vlm_pipeline.py @@ -3,7 +3,7 @@ from io import BytesIO from pathlib import Path from typing import List, Optional, Union, cast -# from docling_core.types import DoclingDocument +from docling_core.types import DoclingDocument from docling_core.types.doc import BoundingBox, DocItem, ImageRef, PictureItem, TextItem from docling_core.types.doc.document import DocTagsDocument from PIL import Image as PILImage @@ -133,7 +133,7 @@ class VlmPipeline(PaginatedPipeline): doctags_doc = DocTagsDocument.from_doctags_and_image_pairs( doctags_list_c, image_list_c ) - conv_res.document.load_from_doctags(doctags_doc) + conv_res.document = DoclingDocument.load_from_doctags(doctags_doc) # If forced backend text, replace model predicted text with backend one if page.size: