Archives/gpt4all
2
0
Fork 0
mirror of https://github.com/nomic-ai/gpt4all synced 2024-11-02 09:40:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

pyllmodel: print specific error message

Browse Source
This commit is contained in:
Cebtenzzre 2023-09-26 09:29:51 -04:00 committed by Adam Treat
parent 34f2ec2b33
commit 4392bf26e0
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
gpt4all-bindings/python/gpt4all/pyllmodel.py
View File

@ -259,12 +259,13 @@ class LLModel:
True if model loaded successfully, False otherwise True if model loaded successfully, False otherwise
""" """
model_path_enc = model_path.encode("utf-8") model_path_enc = model_path.encode("utf-8")
self.model = llmodel.llmodel_model_create(model_path_enc) err = LLModelError()
self.model = llmodel.llmodel_model_create2(model_path_enc, b"auto", ctypes.byref(err))
if self.model is None:
raise ValueError(f"Unable to instantiate model: code={err.code}, {err.message.decode()}")
if self.model is not None:
llmodel.llmodel_loadModel(self.model, model_path_enc) llmodel.llmodel_loadModel(self.model, model_path_enc)
else:
raise ValueError("Unable to instantiate model")
filename = os.path.basename(model_path) filename = os.path.basename(model_path)
self.model_name = os.path.splitext(filename)[0] self.model_name = os.path.splitext(filename)[0]