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

Merge pull request #521 from bretello/main

Browse source 
raise exception when `llama_load_model_from_file` fails
This commit is contained in:
Andrei 2023-07-24 13:09:28 -04:00 committed by GitHub
parent d8a3ddbb1c 8be7d67f7e
commit 07f0f3a386
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
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

@ -423,7 +423,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]