From bd042526a63652a318699e5b471cc3f90a3aef2a Mon Sep 17 00:00:00 2001 From: jane-temcious Date: Thu, 26 Jun 2025 19:50:13 +0530 Subject: [PATCH] Update base_pipeline.py Signed-off-by: jane-temcious --- docling/pipeline/base_pipeline.py | 1 + 1 file changed, 1 insertion(+) diff --git a/docling/pipeline/base_pipeline.py b/docling/pipeline/base_pipeline.py index 4513df7c..466060b5 100644 --- a/docling/pipeline/base_pipeline.py +++ b/docling/pipeline/base_pipeline.py @@ -5,6 +5,7 @@ import traceback from abc import ABC, abstractmethod from collections.abc import Iterable from typing import Any, Callable, List +from docling.utils.gpu_utils import clear_gpu_memory from docling_core.types.doc import NodeItem