Moved keep_backend = True to vlm pipeline

Signed-off-by: Maksym Lysak <[email protected]>
DS4SD · Feb 13, 2025 · 61fce90 · 61fce90
1 parent 9476e6a
commit 61fce90
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/docling/pipeline/base_pipeline.py b/docling/pipeline/base_pipeline.py
@@ -116,10 +116,7 @@ class PaginatedPipeline(BasePipeline):  # TODO this is a bad name.
 
     def __init__(self, pipeline_options: PipelineOptions):
         super().__init__(pipeline_options)
-        self.keep_backend = (
-            True  # For now, need to be able to query for page size post prediction
-        )
-        # self.keep_backend = False
+        self.keep_backend = False
 
     def _apply_on_pages(
         self, conv_res: ConversionResult, page_batch: Iterable[Page]

diff --git a/docling/pipeline/vlm_pipeline.py b/docling/pipeline/vlm_pipeline.py
@@ -42,6 +42,7 @@ class VlmPipeline(PaginatedPipeline):
 
     def __init__(self, pipeline_options: VlmPipelineOptions):
         super().__init__(pipeline_options)
+        self.keep_backend = True
 
         warnings.warn(
             "This API is currently experimental and may change in upcoming versions without notice.",