diff --git a/deploy/python/infer.py b/deploy/python/infer.py
index b167b04ce9c696857bf579d31eb2c479b4783b67..1c7be6373b8afd8a6975fed0f4d46f1fb91aff28 100644
--- a/deploy/python/infer.py
+++ b/deploy/python/infer.py
@@ -865,7 +865,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)