From 09a611b0c9f9ec7abee2b76678afe88751173655 Mon Sep 17 00:00:00 2001
From: duanyanhui <45005871+YanhuiDua@users.noreply.github.com>
Date: Fri, 5 May 2023 10:08:16 +0800
Subject: [PATCH] update npu api (#8172)

---
 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 31e491b12..e920637af 100644
--- a/deploy/python/infer.py
+++ b/deploy/python/infer.py
@@ -868,7 +868,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