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 abc import ABC, abstractmethod
from collections.abc import Iterable from collections.abc import Iterable
from typing import Any, Callable, List from typing import Any, Callable, List
from docling.utils.gpu_utils import clear_gpu_memory
from docling_core.types.doc import NodeItem from docling_core.types.doc import NodeItem