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

Merge branch 'develop' into set_stream_infer-shareExData

Browse Source
This commit is contained in:
Wang Bojun
2023-02-14 14:23:33 +08:00
committed by GitHub
parent 727fdc5863 3aafceed07
commit a0fc0d5e49
14 changed files with 385 additions and 253 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
python/fastdeploy/runtime.py
View File

@@ -195,6 +195,12 @@ class RuntimeOption:
return self._option.set_model_buffer(model_buffer, params_buffer,
model_format)
def set_encryption_key(self, encryption_key):
"""When loading encrypted model, encryption_key is required to decrypte model
:param encryption_key: (str)The key for decrypting model
"""
return self._option.set_encryption_key(encryption_key)
def use_gpu(self, device_id=0):
"""Inference with Nvidia GPU