From 7b2d368ac62f635094d2562a9e973f7b5eec083f Mon Sep 17 00:00:00 2001 From: SWHL Date: Fri, 22 May 2026 14:45:35 +0800 Subject: [PATCH] fix: fixed dml typo --- python/rapidocr/config.yaml | 2 +- .../rapidocr/inference_engine/onnxruntime/provider_config.py | 4 ++-- python/tests/test_files/test_config.yaml | 2 +- 3 files changed, 4 insertions(+), 4 deletions(-) diff --git a/python/rapidocr/config.yaml b/python/rapidocr/config.yaml index 7327e79..30de7a5 100644 --- a/python/rapidocr/config.yaml +++ b/python/rapidocr/config.yaml @@ -35,7 +35,7 @@ EngineConfig: do_copy_in_default_stream: true use_dml: false - dm_ep_cfg: null + dml_ep_cfg: null use_cann: false cann_ep_cfg: diff --git a/python/rapidocr/inference_engine/onnxruntime/provider_config.py b/python/rapidocr/inference_engine/onnxruntime/provider_config.py index 4a5f0f3..18092c9 100644 --- a/python/rapidocr/inference_engine/onnxruntime/provider_config.py +++ b/python/rapidocr/inference_engine/onnxruntime/provider_config.py @@ -59,8 +59,8 @@ class ProviderConfig: return dict(self.cfg.cuda_ep_cfg) def dml_ep_cfg(self) -> Dict[str, Any]: - if self.cfg.dm_ep_cfg is not None: - return self.cfg.dm_ep_cfg + if self.cfg.dml_ep_cfg is not None: + return self.cfg.dml_ep_cfg if self.is_cuda_available(): return self.cuda_ep_cfg() diff --git a/python/tests/test_files/test_config.yaml b/python/tests/test_files/test_config.yaml index 6f6ccec..9a77659 100644 --- a/python/tests/test_files/test_config.yaml +++ b/python/tests/test_files/test_config.yaml @@ -35,7 +35,7 @@ EngineConfig: do_copy_in_default_stream: true use_dml: false - dm_ep_cfg: null + dml_ep_cfg: null use_cann: false cann_ep_cfg: