From 9ea3ac2dcaeaeb010c806facdb159eb80a9d8d9c Mon Sep 17 00:00:00 2001
From: duanyanhui <45005871+YanhuiDua@users.noreply.github.com>
Date: Fri, 5 May 2023 10:23:24 +0800
Subject: [PATCH] update npu api (#8179)

---
 deploy/python/infer.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/deploy/python/infer.py b/deploy/python/infer.py
index b167b04ce..1c7be6373 100644
--- a/deploy/python/infer.py
+++ b/deploy/python/infer.py
@@ -865,7 +865,7 @@ def load_predictor(model_dir,
     elif device == 'NPU':
         if config.lite_engine_enabled():
             config.enable_lite_engine()
-        config.enable_npu()
+        config.enable_custom_device('npu')
     else:
         config.disable_gpu()
         config.set_cpu_math_library_num_threads(cpu_threads)
-- 
GitLab