diff --git a/llama_cpp/llama_cpp.py b/llama_cpp/llama_cpp.py index 423a4a0..87a3b9a 100644 --- a/llama_cpp/llama_cpp.py +++ b/llama_cpp/llama_cpp.py @@ -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]