rixty/KoboldAI-Client
1
1
Fork 0
mirror of https://github.com/KoboldAI/KoboldAI-Client.git synced 2025-06-05 21:59:24 +02:00
Code Issues Projects Releases Wiki Activity

Add more info to custom model error

Browse Source
This commit is contained in:
onesome
2023-04-26 16:36:52 -05:00
parent bbf4963d6e
commit 6776a71532
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
modeling/inference_models/hf.py
View File

@@ -26,9 +26,13 @@ class HFInferenceModel(InferenceModel):
if self.model_name in ["NeoCustom", "GPT2Custom", "TPUMeshTransformerGPTJ", "TPUMeshTransformerGPTNeoX"]: if self.model_name in ["NeoCustom", "GPT2Custom", "TPUMeshTransformerGPTJ", "TPUMeshTransformerGPTNeoX"]:
assert utils.koboldai_vars.custmodpth 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 return utils.koboldai_vars.custmodpth
basename = utils.koboldai_vars.model.replace("/", "_") basename = utils.koboldai_vars.model.replace("/", "_")