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

fix: Correct the way to set GPU for EasyOCR, RapidOCR

Browse Source 
Signed-off-by: Nikos Livathinos <nli@zurich.ibm.com>

Signed-off-by: Christoph Auer <cau@zurich.ibm.com>
This commit is contained in:
Nikos Livathinos 2024-12-10 15:05:00 +00:00
parent 99ccb69a47
commit 6bc1bd2ec4
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
docling/models/easyocr_model.py
View File

@ -48,7 +48,7 @@ class EasyOcrModel(BaseOcrModel):
use_gpu = any( use_gpu = any(
filter( filter(
lambda x: str(x) in device, lambda x: str(x) in device,
[AcceleratorDevice.CUDA, AcceleratorDevice.MPS], ["cuda", "mps"],
) )
) )

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 = AcceleratorDevice.CUDA in device use_cuda = "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