baalajimaestro/llama.cpp
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

raise exception when llama_load_model_from_file fails

Browse source
This commit is contained in:
bretello 2023-07-24 14:42:37 +02:00
parent 231123ee1e
commit 8be7d67f7e
No known key found for this signature in database
GPG key ID: 876AAC6290170FE7
1 changed files with 4 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
llama_cpp/llama_cpp.py
View file

@ -367,7 +367,10 @@ _lib.llama_backend_free.restype = None
def llama_load_model_from_file(
path_model: bytes, params: llama_context_params
) -> llama_model_p:
return _lib.llama_load_model_from_file(path_model, params)
result = _lib.llama_load_model_from_file(path_model, params)
if result is None:
raise Exception(f"Failed to load model from {path_model}")
return result
_lib.llama_load_model_from_file.argtypes = [c_char_p, llama_context_params]