update npu inference api (#9764)

2023-04-20 10:26:37 +08:00 · 2023-04-20 10:26:37 +08:00 · 013870d9bc
parent 4b8e333f10
commit 013870d9bc
1 changed files with 1 additions and 1 deletions
--- a/tools/infer/utility.py
+++ b/tools/infer/utility.py
@ -246,7 +246,7 @@ def create_predictor(args, mode, logger):
                    logger.info("Please keep your paddlepaddle-gpu >= 2.3.0!")

        elif args.use_npu:
-            config.enable_npu()
+            config.enable_custom_device("npu")
        elif args.use_xpu:
            config.enable_xpu(10 * 1024 * 1024)
        else: