This commit is contained in:
Richard Guo 2023-06-02 11:13:26 -04:00 committed by AT
parent fb09f412ff
commit 46986d5f26

View File

@ -75,6 +75,9 @@ llmodel.llmodel_model_create.restype = ctypes.c_void_p
llmodel.llmodel_model_create2.argtypes = [ctypes.c_char_p, ctypes.c_char_p, ctypes.POINTER(LLModelError)] llmodel.llmodel_model_create2.argtypes = [ctypes.c_char_p, ctypes.c_char_p, ctypes.POINTER(LLModelError)]
llmodel.llmodel_model_create2.restype = ctypes.c_void_p llmodel.llmodel_model_create2.restype = ctypes.c_void_p
llmodel.llmodel_model_destroy.argtypes = [ctypes.c_void_p]
llmodel.llmodel_model_destroy.restype = None
llmodel.llmodel_loadModel.argtypes = [ctypes.c_void_p, ctypes.c_char_p] llmodel.llmodel_loadModel.argtypes = [ctypes.c_void_p, ctypes.c_char_p]
llmodel.llmodel_loadModel.restype = ctypes.c_bool llmodel.llmodel_loadModel.restype = ctypes.c_bool
llmodel.llmodel_isModelLoaded.argtypes = [ctypes.c_void_p] llmodel.llmodel_isModelLoaded.argtypes = [ctypes.c_void_p]
@ -124,7 +127,7 @@ class LLModel:
self.model_name = None self.model_name = None
def __del__(self): def __del__(self):
if self.model is not None and llmodel is not None: if self.model is not None:
llmodel.llmodel_model_destroy(self.model) llmodel.llmodel_model_destroy(self.model)
def load_model(self, model_path: str) -> bool: def load_model(self, model_path: str) -> bool: