diff --git a/deploy/python/infer.py b/deploy/python/infer.py
index 31e491b12783a80aebfdbc09866b114063387717..e920637af3dc43aa5f1b32fcea2792c1554cebd5 100644
--- a/deploy/python/infer.py
+++ b/deploy/python/infer.py
@@ -868,7 +868,7 @@ def load_predictor(model_dir,
     elif device == 'NPU':
         if config.lite_engine_enabled():
             config.enable_lite_engine()
-        config.enable_npu()
+        config.enable_custom_device('npu')
     else:
         config.disable_gpu()
         config.set_cpu_math_library_num_threads(cpu_threads)