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

Add decrypt function to load encrypted model

Browse Source
This commit is contained in:
felixhjh
2023-02-09 14:04:10 +00:00
parent 05be955187
commit 942cee83d7
5 changed files with 45 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
fastdeploy/runtime/option_pybind.cc
View File

@@ -35,6 +35,7 @@ void BindOption(pybind11::module& m) {
.def(pybind11::init())
.def("set_model_path", &RuntimeOption::SetModelPath)
.def("set_model_buffer", &RuntimeOption::SetModelBuffer)
.def("set_encryption_key", &RuntimeOption::SetEncryptionKey)
.def("use_gpu", &RuntimeOption::UseGpu)
.def("use_cpu", &RuntimeOption::UseCpu)
.def("use_rknpu2", &RuntimeOption::UseRKNPU2)