Add decrypt function to load encrypted model

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

View File

@@ -187,6 +187,13 @@ 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