mirror of
https://github.com/KoboldAI/KoboldAI-Client.git
synced 2025-06-05 21:59:24 +02:00
Fix for --path
This commit is contained in:
@@ -338,9 +338,11 @@ class HFInferenceModel(InferenceModel):
|
||||
Returns a string of the model's path locally, or None if it is not downloaded.
|
||||
If ignore_existance is true, it will always return a path.
|
||||
"""
|
||||
if os.path.exists(self.path):
|
||||
return self.path
|
||||
|
||||
if self.model_name in ["NeoCustom", "GPT2Custom", "TPUMeshTransformerGPTJ", "TPUMeshTransformerGPTNeoX"]:
|
||||
model_path = utils.koboldai_vars.custmodpth
|
||||
model_path = self.path
|
||||
assert model_path
|
||||
|
||||
# Path can be absolute or relative to models directory
|
||||
|
Reference in New Issue
Block a user