fix: Raise exceptions when llama model or context fails to load

abetlen · abetlen · commit dd22010e8526 · 2024-02-22T00:09:45.000-05:00
diff --git a/llama_cpp/_internals.py b/llama_cpp/_internals.py
@@ -51,6 +51,9 @@ def __init__(
             self.path_model.encode("utf-8"), self.params
         )
 
+        if self.model is None:
+            raise ValueError(f"Failed to load model from file: {path_model}")
+
     def __del__(self):
         if self.model is not None and self._llama_free_model is not None:
             self._llama_free_model(self.model)
@@ -258,6 +261,9 @@ def __init__(
             self.model.model, self.params
         )
 
+        if self.ctx is None:
+            raise ValueError("Failed to create llama_context")
+
     def __del__(self):
         if self.ctx is not None and self._llama_free is not None:
             self._llama_free(self.ctx)
