Update base_pipeline.py

Signed-off-by: jane-temcious <jane.temcious@iterate.ai>
This commit is contained in:
jane-temcious 2025-06-26 19:50:13 +05:30 committed by GitHub
parent 1fdfb2bc50
commit bd042526a6
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -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