Merge pull request abetlen#521 from bretello/main

abetlen · web-flow · commit 07f0f3a3860a · 2023-07-24T13:09:28.000-04:00
raise exception when `llama_load_model_from_file` fails
diff --git a/llama_cpp/llama_cpp.py b/llama_cpp/llama_cpp.py
@@ -423,7 +423,10 @@ def llama_backend_free():
 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]
