apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-06 00:57:33 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
Zheng-Bicheng
2023-03-07 23:02:43 +08:00
parent b5fb9fad4e
commit 1e56d5069c
1 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
python/fastdeploy/runtime.py
View File

@@ -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):