update npu inference api

6a55aac3 · duanyanhui · cuicheng01 · a89269e5 · 6a55aac3
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

deploy/utils/predictor.py deploy/utils/predictor.py +1 -1

未找到文件。
--- 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: