mirror of
https://github.com/RapidAI/RapidOCR.git
synced 2026-11-10 21:10:55 +08:00
fix: fixed dml typo
This commit is contained in:
3 files changed
+4
-4
No files matched your search
@@ -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:
|
||||
|
||||
@@ -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()
|
||||
|
||||
@@ -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:
|
||||
|
||||
Reference in new issue
Block a user