From 6776a71532e8da591ecf0eab01e1934b6a25f069 Mon Sep 17 00:00:00 2001 From: onesome Date: Wed, 26 Apr 2023 16:36:52 -0500 Subject: [PATCH] Add more info to custom model error --- modeling/inference_models/hf.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/modeling/inference_models/hf.py b/modeling/inference_models/hf.py index d7d187c2..f795395e 100644 --- a/modeling/inference_models/hf.py +++ b/modeling/inference_models/hf.py @@ -26,9 +26,13 @@ class HFInferenceModel(InferenceModel): if self.model_name in ["NeoCustom", "GPT2Custom", "TPUMeshTransformerGPTJ", "TPUMeshTransformerGPTNeoX"]: assert utils.koboldai_vars.custmodpth - assert os.path.exists(utils.koboldai_vars.custmodpth) - print("CUSTMODPATH") + try: + assert os.path.exists(utils.koboldai_vars.custmodpth) + except AssertionError: + logger.error(f"Custom model at '{utils.koboldai_vars.custmodpth}' doesn't seem to exist") + raise + return utils.koboldai_vars.custmodpth basename = utils.koboldai_vars.model.replace("/", "_")