From f8fdc5fd9812db631bb786c60ea335d0fd07c395 Mon Sep 17 00:00:00 2001 From: duanyanhui Date: Thu, 4 May 2023 10:59:16 +0800 Subject: [PATCH] update npu inference api --- deploy/utils/predictor.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deploy/utils/predictor.py b/deploy/utils/predictor.py index 948b1859..c13a4080 100644 --- 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: -- GitLab