update npu inference api (PaddlePaddle#9765)

truongnn1106 · Apr 20, 2023 · ce65745 · ce65745
1 parent d496c8e
commit ce65745
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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: