mirrors/docling
1
0
Fork 0
mirror of https://github.com/DS4SD/docling.git synced 2025-08-02 07:22:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update custom_convert.py

Browse Source 
Added the missing AcceleratorDevice and AcceleratorOptions functions in the imports and changed Device in the code to the correct AcceleratorDevice

Signed-off-by: Iacopo Ghinassi <45108036+Ighina@users.noreply.github.com>
This commit is contained in:
Iacopo Ghinassi 2025-01-13 17:05:27 +00:00 committed by GitHub
parent 1976584be1
commit 188789b221
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
docs/examples/custom_convert.py
View File

@ -5,7 +5,7 @@ from pathlib import Path
from docling.backend.pypdfium2_backend import PyPdfiumDocumentBackend from docling.backend.pypdfium2_backend import PyPdfiumDocumentBackend
from docling.datamodel.base_models import InputFormat from docling.datamodel.base_models import InputFormat
from docling.datamodel.pipeline_options import PdfPipelineOptions from docling.datamodel.pipeline_options import AcceleratorDevice, AcceleratorOptions, PdfPipelineOptions
from docling.document_converter import DocumentConverter, PdfFormatOption from docling.document_converter import DocumentConverter, PdfFormatOption
from docling.models.ocr_mac_model import OcrMacOptions from docling.models.ocr_mac_model import OcrMacOptions
from docling.models.tesseract_ocr_cli_model import TesseractCliOcrOptions from docling.models.tesseract_ocr_cli_model import TesseractCliOcrOptions
@ -76,7 +76,7 @@ def main():
pipeline_options.table_structure_options.do_cell_matching = True pipeline_options.table_structure_options.do_cell_matching = True
pipeline_options.ocr_options.lang = ["es"] pipeline_options.ocr_options.lang = ["es"]
pipeline_options.accelerator_options = AcceleratorOptions( pipeline_options.accelerator_options = AcceleratorOptions(
num_threads=4, device=Device.AUTO num_threads=4, device=AcceleratorDevice.AUTO
) )
doc_converter = DocumentConverter( doc_converter = DocumentConverter(