mirror of
https://github.com/DS4SD/docling.git
synced 2025-07-26 20:14:47 +00:00
Update base_pipeline.py
Signed-off-by: jane-temcious <jane.temcious@iterate.ai>
This commit is contained in:
parent
1fdfb2bc50
commit
bd042526a6
@ -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
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user