model creation is failing... debugging

This commit is contained in:
Richard Guo 2023-06-01 11:27:50 -04:00 committed by AT
parent 9f203c211f
commit 5490af5a2c

View File

@ -148,10 +148,14 @@ class LLModel:
model_path_enc = model_path.encode("utf-8") model_path_enc = model_path.encode("utf-8")
build_var = "auto".encode("utf-8") build_var = "auto".encode("utf-8")
self.model = llmodel.llmodel_model_create2(model_path_enc, build_var, None) self.model = llmodel.llmodel_model_create2(model_path_enc, build_var, None)
llmodel.llmodel_loadModel(self.model, model_path_enc)
if self.model is not None:
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]
if llmodel.llmodel_isModelLoaded(self.model): if llmodel.llmodel_isModelLoaded(self.model):
return True return True
else: else: