mirrors/docling
1
0
Fork 0
mirror of https://github.com/DS4SD/docling.git synced 2025-07-27 04:24:45 +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(
filter(
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
device = decide_device(accelerator_options.device)
use_cuda = AcceleratorDevice.CUDA in device
use_cuda = "cuda" in device
use_dml = accelerator_options.device == AcceleratorDevice.AUTO
intra_op_num_threads = accelerator_options.num_threads