mirrors/docling
1
0
Fork 0
mirror of https://github.com/DS4SD/docling.git synced 2025-07-27 04:24:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

Flipped keep_backend to True for vlm_pipeline assembly to work

Browse Source 
Signed-off-by: Maksym Lysak <mly@zurich.ibm.com>
This commit is contained in:
Maksym Lysak 2025-01-16 16:51:27 +01:00
parent 9901729d8c
commit f6d123a01c
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
docling/pipeline/base_pipeline.py
View File

@ -116,7 +116,7 @@ class PaginatedPipeline(BasePipeline): # TODO this is a bad name.
def __init__(self, pipeline_options: PipelineOptions):
super().__init__(pipeline_options)
self.keep_backend = False
self.keep_backend = True
def _apply_on_pages(
self, conv_res: ConversionResult, page_batch: Iterable[Page]
@ -200,12 +200,12 @@ class PaginatedPipeline(BasePipeline): # TODO this is a bad name.
return conv_res
def _unload(self, conv_res: ConversionResult) -> ConversionResult:
# for page in conv_res.pages:
# if page._backend is not None:
# page._backend.unload()
for page in conv_res.pages:
if page._backend is not None:
page._backend.unload()
# if conv_res.input._backend:
# conv_res.input._backend.unload()
if conv_res.input._backend:
conv_res.input._backend.unload()
return conv_res
def _determine_status(self, conv_res: ConversionResult) -> ConversionStatus: