update

Venonary · Mar 7, 2023 · 1e56d50 · 1e56d50
1 parent b5fb9fa
commit 1e56d50
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/python/fastdeploy/runtime.py b/python/fastdeploy/runtime.py
@@ -16,7 +16,6 @@
 import numpy as np
 from . import ModelFormat
 from . import c_lib_wrap as C
-from . import rknpu2
 
 
 class Runtime:
@@ -253,8 +252,8 @@ def use_cpu(self):
         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):