raise exception when llama_load_model_from_file fails

bretello · bretello · commit 8be7d67f7e64 · 2023-07-24T14:42:37.000+02:00
diff --git a/llama_cpp/llama_cpp.py b/llama_cpp/llama_cpp.py
@@ -367,7 +367,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]
