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

Merge branch 'nli/performance' of github.com:DS4SD/docling into cau/layout-processing-improvement

Browse Source
This commit is contained in:
Christoph Auer 2024-12-10 15:57:39 +01:00
commit 5497ec8a66
2 changed files with 8 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
docling/models/easyocr_model.py
View File

@ -45,10 +45,13 @@ class EasyOcrModel(BaseOcrModel):
if self.options.use_gpu: if self.options.use_gpu:
device = decide_device(accelerator_options.device) device = decide_device(accelerator_options.device)
# Enable easyocr GPU if running on CUDA, MPS # Enable easyocr GPU if running on CUDA, MPS
use_gpu = device in [ use_gpu = any(
AcceleratorDevice.CUDA, filter(
AcceleratorDevice.MPS, lambda x: str(x) in device,
] [AcceleratorDevice.CUDA, AcceleratorDevice.MPS],
)
)
self.reader = easyocr.Reader( self.reader = easyocr.Reader(
lang_list=self.options.lang, lang_list=self.options.lang,
gpu=use_gpu, gpu=use_gpu,

2
docling/models/rapid_ocr_model.py
View File

@ -42,7 +42,7 @@ class RapidOcrModel(BaseOcrModel):
# Decide the accelerator devices # Decide the accelerator devices
device = decide_device(accelerator_options.device) device = decide_device(accelerator_options.device)
use_cuda = device == AcceleratorDevice.CUDA use_cuda = AcceleratorDevice.CUDA in device
use_dml = accelerator_options.device == AcceleratorDevice.AUTO use_dml = accelerator_options.device == AcceleratorDevice.AUTO
intra_op_num_threads = accelerator_options.num_threads intra_op_num_threads = accelerator_options.num_threads