update npu inference api

2023-05-04 10:59:16 +08:00 · 2023-05-04 10:59:16 +08:00 · f8fdc5fd98
parent e7bef51f9e
commit f8fdc5fd98
1 changed files with 1 additions and 1 deletions
--- a/deploy/utils/predictor.py
+++ b/deploy/utils/predictor.py
@ -63,7 +63,7 @@ class Predictor(object):
        if args.get("use_gpu", False):
            config.enable_use_gpu(args.gpu_mem, 0)
        elif args.get("use_npu", False):
-            config.enable_npu()
+            config.enable_custom_device('npu')
        elif args.get("use_xpu", False):
            config.enable_xpu()
        else: