diff --git a/python/fastdeploy/runtime.py b/python/fastdeploy/runtime.py
index 4980db52e..f7480de4b 100755
--- a/python/fastdeploy/runtime.py
+++ b/python/fastdeploy/runtime.py
@@ -16,7 +16,6 @@ import logging
 import numpy as np
 from . import ModelFormat
 from . import c_lib_wrap as C
-from . import rknpu2
 
 
 class Runtime:
@@ -253,8 +252,8 @@ class RuntimeOption:
         return self._option.use_cpu()
 
     def use_rknpu2(self,
-                   rknpu2_name=rknpu2.CpuName.RK3588,
-                   rknpu2_core=rknpu2.CoreMask.RKNN_NPU_CORE_0):
+                   rknpu2_name=C.CpuName.RK356X,
+                   rknpu2_core=C.CoreMask.RKNN_NPU_CORE_AUTO):
         return self._option.use_rknpu2(rknpu2_name, rknpu2_core)
 
     def use_sophgo(self):